Water surface floating object automatic detection method and device for urban river regulation

Abstract

The invention discloses a water surface floating object automatic detection method and device for urban river regulation. The method comprises the steps if automatically extracting an image from a water surface monitoring video stream, and enhancing the image based on a self-adaptive contrast ratio; clustering the size of a target detection object in the image to obtain size distribution of a detected target; constructing an improved Faster R-CNN network model, determining a fused feature layer according to the size of a target detection object and the size of a receptive field of each layer of the convolutional neural network model, and fusing features extracted across the feature layers in a splicing mode; determining the size and the length-width ratio of an RPN network anchor frame according to a clustering result, and generating suggested areas of targets of different sizes; and optimizing network model parameters through an iterative method and realizing detection of water floaters. The invention can achieve the automatic recognition and marking of the floaters in the video stream, solves problems that the detection accuracy is low and the generalization capability is poor, can achieve the automatic detection of the floaters, and can be used for the automatic supervision of an actual project, such as an urban river channel.

Description

technical field [0001] The invention relates to an automatic detection method and device for floating objects on the water surface facing urban river channel supervision, and belongs to the fields of artificial intelligence and computer vision. Background technique [0002] With the rapid economic and social development, water ecology and water environment issues have become one of the people's livelihood issues that the masses are most concerned about. The accumulation of floating objects on the water surface not only affects the quality of the urban living environment, but also causes water pollution and damage to the ecological balance, and even threatens the safety of drinking water. In order to solve complex water environment problems, the river chief system proposes to strengthen water environment governance, and relevant departments can report and respond quickly to the floating objects on the water surface in a timely manner, which requires intelligent detection of f...

AI Technical Summary

Problems solved by technology

[0004] Purpose of the invention: In view of the deficiencies in the prior art, the purpose of the present invention is to provide an automatic detection method and device for floating objects on the water surface for urban river supervision, so as to overcome the poor generalization ability and low detection accuracy of the traditional floating object detection methods and other issues; and based on the size distribution of floating objects, the network model is improved, and the features extracted by the network across feature layers are fused to automatically generate suggested areas for objects of different sizes, further improving the accuracy of floating object detection

Images