Fig. 1. YOLOv3 network structure diagram

Fig. 2. Residual unit and SE-residual unit

Fig. 3. Improved feature fusion network structure. (a) FPN; (b) bottom-up path aggregation

Fig. 4. YOLO-Mask algorithm network structure

Fig. 5. Example of dataset images

Fig. 6. Comparison of detection results of different algorithms. (a)(b)(c) SSD detection results; (d)(e)(f) Faster R-CNN detection results; (g)(h)(i) YOLOv3 detection results; (j)(k)(l) YOLO-Mask detection results

Fig. 7. Single target P-R curves of YOLO-Mask algorithm. (a) Right_mask; (b) wrong_mask; (c) no_mask