what is the function of the file coco_supervision.txt? i want to train own datastet, need i generate the same file.

Megvii-BaseDetection / DenseTeacher

DenseTeacher: Dense Pseudo-Label for Semi-supervised Object Detection

Apache License 2.0

120 stars 12 forks source link

what is the function of the file coco_supervision.txt? i want to train own datastet, need i generate the same file. #21

Closed liuhaolinwen closed 1 year ago

liuhaolinwen commented 1 year ago

i don't konw the function of the coco_supervision.txt, i want to train own datastet, need i generate the same file, what should i do?

ZRandomize commented 1 year ago

This file is used in dataset division here. It distinguish an COCO image belongs to supervised set or unsupervised set. If you are using your own data, please refer to this issue.

liuhaolinwen commented 1 year ago

How to generate coco_supervision file? do you have any script?

ZRandomize commented 1 year ago

no, this is copied from unbiased teacher. You can simply generate it using: size=len(dataset) labeled_idx=np.random.choice(size,int(size*split_percentage/100),replace=False) unlabeled_idx=list(set(range(size)) - set(labeled_idx))