CenterNet2 datset issue

Hi,when I train the model on my own dataset,I find total number of the dataset which is shown in the picture is not equal to the actual number of dataset，what lead to this problem。 As shown in the picture, It says total picture number is 83810,but actually picture number is 49105。

Mar 18 '21 08:03 qdd1234

Hi, That's interesting. Can you try to manually load the annotation file and check the number of images? Can you check your config file only contains that single dataset in DATASETS.TRAIN?

Mar 18 '21 15:03 xingyizhou

Hi,How can I find the savd model? I find a folder named output,but in the folder i can not find .pth.

Mar 19 '21 08:03 qdd1234

@qdd1234, I used register_coco_instances to register my custom dataset but when I pass it to train_net.py it shows that my dataset has not been registered. Could you show me how you can pass your own dataset to train_net.py. Thank you

Mar 19 '21 12:03 DatDoc

@qdd1234, I used register_coco_instances to register my custom dataset but when I pass it to train_net.py it shows that my dataset has not been registered. Could you show me how you can pass your own dataset to train_net.py. Thank you

My dataset format is VOC. I directly change the voc in the detecton2 including class_num,img_path as so on

Mar 19 '21 12:03 qdd1234

@qdd1234 But you are using train_net.py to train, right? Thanks for the quick reply

Mar 19 '21 12:03 DatDoc

@qdd1234 But you are using train_net.py to train, right? Thanks for the quick reply

Yes

Mar 19 '21 12:03 qdd1234

Could you show me the configuration for your custom dataset?

Mar 19 '21 12:03 DatDoc

I directly changed the configuration of the VOC data set in Detectron2, including the VOC data category, the original VOC path, and the file name when reading the TXT file to my own file

Mar 19 '21 13:03 qdd1234