arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2204.07437cs.CVcs.LG

Transfer Learning for Instance Segmentation of Waste Bottles using Mask R-CNN Algorithm

University of Reading(雷丁大学)

Punitha Jaikumar, Remy Vandaele, Varun Ojha

英文摘要:

This paper proposes a methodological approach with a transfer learning scheme for plastic waste bottle detection and instance segmentation using the \textit{mask region proposal convolutional neural network} (Mask R-CNN). Plastic bottles constitute one of the major pollutants posing a serious threat to the environment both in oceans and on land. The automated identification and segregation of bottles can facilitate plastic waste recycling. We prepare a custom-made dataset of 192 bottle images with pixel-by pixel-polygon annotation for the automatic segmentation task. The proposed transfer learning scheme makes use of a Mask R-CNN model pre-trained on the Microsoft COCO dataset. We present a comprehensive scheme for fine-tuning the base pre-trained Mask-RCNN model on our custom dataset. Our final fine-tuned model has achieved 59.4 \textit{mean average precision} (mAP), which corresponds to the MS COCO metric. The results indicate a promising application of deep learning for detecting waste bottles.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑