Floating object identification method based on improved SSD model and storage medium

The invention provides a floating object identification method based on an improved SSD model, and the method comprises the steps: inputting image data into an optimized SSD model to identify a target floating object in an image, and a training process of the optimized SSD model comprises the steps: collecting original water surface image data, and carrying out the disordering processing, so as to obtain an original sample set; performing image enhancement processing on original samples in the original sample set to obtain an expanded sample set; floating object information is manually marked on the expanded samples in the expanded sample set; and carrying out model training on a pre-trained optimized SSD model by adopting sample data in the expansion sample set, wherein the optimized SSD model comprises a plurality of small target detection layers. According to the floating object identification method based on the improved SSD model, through the enhancement processing process of the water surface image data