The invention discloses a water surface floating
object detection and identification method based on an improved YOLOv3 identification model, and relates to the technical field of
computer vision. The method comprises the following steps: pre-collecting water surface floating object data, carrying out the enhancement and amplification of image data through
geometric transformation and
color transformation, marking the floating objects in the data, obtaining a water surface drifting object
data set, and splitting the water surface drifting object
data set into a
training set and a
test set; constructing an improved YOLOv3
network model, and training the improved YOLOv3
network model by adopting the water surface drifting object
training set; constructing a water surface drifting object
test set according to the water surface drifting object data image, and detecting and identifying the water surface drifting object
test set by using the trained improved YOLOv3
network model. The improved YOLOv3 has strong generalization ability, occupies small storage space and
video memory space, improves detection and identification accuracy, can ensure real-time performance, and can realize accurate and rapid monitoring and identification of water surface drifts in
client equipment with limited computing power and memory.