Search papers, labs, and topics across Lattice.
This paper introduces DropClick, a semi-automated segmentation tool that leverages single-click inputs to generate pseudo-labels, significantly reducing the labor involved in annotating agricultural robotic datasets. Evaluated on the SB20 and BUP20 datasets, DropClick achieves competitive mean Intersection over Union (mIoU) scores of 70.0 and 72.6, respectively, while maintaining high performance even with up to 50% of user inputs missing. The method demonstrates that substantial user input can be saved without compromising segmentation quality, validating its effectiveness as a pseudo-labeling approach in semi-supervised learning contexts.
DropClick can maintain high segmentation performance with minimal user input, saving over 46% of annotation effort while still achieving competitive results.
Labelling vision datasets, especially for segmentation tasks, is a laborious and costly process that stymies novel developments in agricultural robotics. In this paper, we present DropClick, a click-guided segmentation tool that simplifies the annotation process. Our system utilises single-click inputs on objects to generate pseudo-labels, which can replace manual annotations. DropClick stands out as it is a semi-automated approach and does not require a click for every object in the scene. It can therefore further reduce the required amount of user input drastically. We evaluate our method on two challenging agricultural robotic datasets, SB20 and BUP20 for plant and fruit segmentation, respectively. DropClick is first trained on a small subset of just 5 images from the original training data. This DropClick model can then be deployed as a one-click segmentation system and achieves comparable or higher performance than other one-click methods achieving an mIoU of 70.0 and 72.6 points, for SB20 and BUP20 respectively. DropClick then excels at maintaining high performance when clicks are not given (e.g. dropped); when 50% of the clicks are missing it still maintains an mIoU of 68.9 and 71.3 points, for SB20 and BUP20 respectively. We validate DropClick as a pseudo-labelling approach by taking its outputs to train a Mask2Former instance-based segmentation model in a semi-supervised manner. In this process, partially removing user input from DropClick yields similar high performance when compared to providing all clicks, at 70.1 vs 70.7 points AP50 for SB20 and no difference for BUP20 at 77.0 for both models; at the same time saving 46.3% of total input for SB20 and 31.9% for BUP20.
