what about custom dataset training?

[orkunozturk]

Hi, thanks for your work.

I wonder how can i train and test on my custom dataset. Should i create reference annotation data? Can you share the related script to generate this file?

[saswat0]

Hi @hero-y
It'd be great if you could share the above script or explain its significance.

[hero-y]

Hi, I don't try our method on the custom dataset, but I think it isn't difficult. First, you should prepare your custom dataset with the coco format. Then, you don't need to prepare the ref_ann_file (refer to voc config). If the ref_ann_file doesn't exist, the code will choose one class-object annotation from the whole annotation as the reference (refer to datasets/one_shot_voc).

[szgy66]

Hi, thanks for your work.

I wonder how can i train and test on my custom dataset. Should i create reference annotation data? Can you share the related script to generate this file?

Hello, Had you train and test on your custom dataset?