Laser & Optoelectronics Progress, Volume. 57, Issue 12, 121007(2020)

Application of Improved Faster RCNN Model for Foreign Fiber Identification in Cotton

Yuhong Du1,2、*, Chaoqun Dong1,2, Di Zhao1,2, Weijia Ren1,2, and Wenchao Cai3
Author Affiliations
  • 1College of Mechanical Engineering, Tianjin Polytechnic University, Tianjin 300387, China
  • 2Tianjin Key Laboratory of Advanced Mechatronics Equipment Technology, Tianjin 300387, China
  • 3Beijing Daheng Image Vision Co., Ltd., Beijing 100085, China
  • show less

    Generally, deep learning methods are used to identify and classify the foreign fibers in cotton. First, a target recognition framework is adopted based on Faster RCNN to develop a foreign fiber dataset according to the characteristics of foreign fiber size and shape diversity. Next, the original VGG16 is replaced by RseNet-50 as the feature extraction network in the foreign fiber classification model, and the size of the mark box is improved using the k-means++ clustering algorithm. Subsequently, the model is trained to identify and classify the foreign fibers in cotton. The trained model achieves an accuracy rate of 94.24%, a precision of 98.16%, a recall rate of 95.93%, and an F1 score of 0.970 with respect to the verification set. When compared with the original model, the recognition effect is observed to improve in case of small sizes, large aspect ratios, and dense occurrences when the proposed model is used. Furthermore, the accuracy, precision, recall rate, and F1 score of the proposed model improve by 3.21%, 0.90%, 2.51%, and 0.017, respectively, when compared with those of the original model.

    Tools

    Get Citation

    Copy Citation Text

    Yuhong Du, Chaoqun Dong, Di Zhao, Weijia Ren, Wenchao Cai. Application of Improved Faster RCNN Model for Foreign Fiber Identification in Cotton[J]. Laser & Optoelectronics Progress, 2020, 57(12): 121007

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Sep. 23, 2019

    Accepted: Oct. 30, 2019

    Published Online: Jun. 3, 2020

    The Author Email: Du Yuhong (DYH202@163.com)

    DOI:10.3788/LOP57.121007

    Topics