Correcting Pseudo Labels in Semi Supervised Object Detection with SAM

Pengyu Wang,Jianmin Li,Wenbo Ding,Jiachen Zhong,Jianyong Ai
DOI: https://doi.org/10.1109/icme57554.2024.10687720
2024-01-01
Abstract:Pseudo label method is a simple and effective method in semi supervised object detection(SSOD). However, pseudo labels inevitably contain noise, which significantly affects the training efficiency and accuracy of the SSOD model. Conventional pseudo label filtering schemes only consider the category accuracy of pseudo labels, but cannot balance the position quality. The current methods for optimizing pseudo labels also have drawbacks such as relying on the performance of the trained model. To mitigate this problem, we propose a basic process for correcting pseudo labels in SSOD using segment anything model (SAM), which can be simply applied to various existing SSOD models. Moreover, we optimize a specific method of generating pseudo labels using the mask obtained from SAM. We apply our method to several representative models, experiment results show that our method can improve the quality of the obtained pseudo labels and model performance.
What problem does this paper attempt to address?