Fast and Accurate Fish Detection Design with Improved YOLO-v3 Model and Transfer Learning

被引:2
作者
Raza, Kazim [1 ]
Hong, Song [1 ]
机构
[1] Zhejiang Univ, Dept Ocean Sci & Engn, Ocean Opto Elect & Automat Lab, Hangzhou 310058, Zhejiang, Peoples R China
关键词
Deep learning; computer vision; transfer learning; improved YOLOv3; anchor box; custom dataset;
D O I
10.14569/ijacsa.2020.0110202
中图分类号
TP301 [理论、方法];
学科分类号
081202 ;
摘要
Object Detection is one of the problematic Computer Vision (CV) problems with countless applications. We proposed a real-time object detection algorithm based on Improved You Only Look Once version 3 (YOLOv3) for detecting fish. The demand for monitoring the marine ecosystem is increasing day by day for a vigorous automated system, which has been beneficial for all of the researchers in order to collect information about marine life. This proposed work mainly approached the CV technique to detect and classify marine life. In this paper, we proposed improved YOLOv3 by increasing detection scale from 3 to 4, apply k-means clustering to increase the anchor boxes, novel transfer learning technique, and improvement in loss function to improve the model performance. We performed object detection on four fish species custom datasets by applying YOLOv3 architecture. We got 87.56% mean Average Precision (mAP). Moreover, comparing to the experimental analysis of the original YOLOv3 model with the improved one, we observed the mAP increased from 87.17% to 91.30. It showed that improved version outperforms than the original YOLOv3 model.
引用
收藏
页码:7 / 16
页数:10
相关论文
共 17 条
  • [1] Design of Pin-Point Gate Injection Mould for Shells of Earplugs
    Feng, Xian
    Yang, Min
    Zou, Min
    [J]. ADVANCED MANUFACTURING AND AUTOMATION VII, 2018, 451 : 53 - 61
  • [2] Fast R-CNN
    Girshick, Ross
    [J]. 2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, : 1440 - 1448
  • [3] He KM, 2017, IEEE I CONF COMP VIS, P2980, DOI [10.1109/TPAMI.2018.2844175, 10.1109/ICCV.2017.322]
  • [4] Hubel D.H, 1961, RECEPTIVE FIELDS BIN, P151
  • [5] Joseph RK, 2016, CRIT POL ECON S ASIA, P1
  • [6] Lantsova Ekaterina, 2015, AUTOMATIC RECOGNITIO, P49
  • [7] Larsen R, 2009, LECT NOTES COMPUT SC, V5575, P745
  • [8] SSD: Single Shot MultiBox Detector
    Liu, Wei
    Anguelov, Dragomir
    Erhan, Dumitru
    Szegedy, Christian
    Reed, Scott
    Fu, Cheng-Yang
    Berg, Alexander C.
    [J]. COMPUTER VISION - ECCV 2016, PT I, 2016, 9905 : 21 - 37
  • [9] Ogunlana S.O., 2015, FISH CLASSIFICATION, P75
  • [10] Pedersen M., 2019, BRACKISH DATASET KAG