Zhihao Chen
Zhihao Chen
In./segmentor/trainer_contrastive.py I think #138 pixel_queue_ptr[lb] = (pixel_queue_ptr[lb] + 1) % self.memory_size should be #138 pixel_queue_ptr[lb] = (pixel_queue_ptr[lb] + K) % self.memory_size
Code bug
In trainer_contrastive.py line129 feat = this_feat[:, perm[:K]] should be idxs_perm = idxs[perm] feat = this_feat[:,idxs_perm[:K]].squeeze() Otherwise, it can not select the current points.
1.I want to know how to generate scribble labels when training? Use skeleton operator? Can you provide those codes? 2.When testing, do you label the scribbles with handcraft way every...