• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

Computer Engineering & Science ›› 2022, Vol. 44 ›› Issue (11): 2003-2009.

• Graphics and Images • Previous Articles     Next Articles

Garbage detection based on Mask R-CNN

ZHANG Rui-ping1,NING Qian1,2,LEI Yin-jie1,CHEN Bing-cai3   

  1. (1.College of Electronics and Information Engineering,Sichuan University,Chengdu 610065;
    2.College of Physics and Electronic Engineering,Xinjiang Normal University,Urumqi  830054;
    3.College of Computer Science and Technology,Dalian University of Technology,Dalian 116024,China)
  • Received:2021-03-19 Revised:2021-08-17 Accepted:2022-11-25 Online:2022-11-25 Published:2022-11-25

Abstract: In recent years, People pay more and more attention to garbage classification and recycling, but garbage classification consumes a lot of manpower and material resources and the sorting efficiency is low.  To solve the problem that the garbage detection method based on rectangular bounding box is not effective enough when applied to multi-classification environment, a garbage detection method based on improved Mask R-CNN is proposed. Instead of the traditional ResNet, this method uses the improved ResNeXt101 as the backbone network for feature extraction, which improves the accuracy of object detection and the accuracy of background boundary segmentation. Experimental results show that compared with the traditional Mask R-CNN model, the proposed model’s average classification accuracy is 91.1%, improved by 2.35%. Finally, the experimental comparison with the current popular object detection algorithms shows that the classification accuracy and segmentation accuracy of the proposed algorithm are excellent, which proves the feasibility and effectiveness of the proposed method in the garbage detection task.

Key words: garbage classification, target detection, regions with CNN features