search

xingyizhou / CenterNet2

Two-stage CenterNet
Apache License 2.0
1.2k stars 188 forks source link

datset issue #7

Open qdd1234 opened 3 years ago

qdd1234 commented 3 years ago

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。 image As shown in the picture, It says total picture number is 83810,but actually picture number is 49105。

xingyizhou commented 3 years ago

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?

qdd1234 commented 3 years ago

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

DatDoc commented 3 years ago

@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

qdd1234 commented 3 years ago

@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

DatDoc commented 3 years ago

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

qdd1234 commented 3 years ago

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

Yes

DatDoc commented 3 years ago

Could you show me the configuration for your custom dataset?

qdd1234 commented 3 years ago

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 image