search

xingyizhou / CenterNet2

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

How to train in VOC #3

Open qdd1234 opened 3 years ago

qdd1234 commented 3 years ago

📚 Documentation

Hi, Thanks for your wonderful work,Could you teach me how to train in VOC2012?

xingyizhou commented 3 years ago

Thank you for your interest. I haven't tried to train on VOC with this code. I guess you can follow the instructions in the original detectron2. I.e., place your data this way and prepare a config like this.

qdd1234 commented 3 years ago

Hi ，If I want to train the model on my own dataste，How can i do？

qdd1234 commented 3 years ago

Hi When I train model on my own dataset,It raises an error which is shown in picture，could you tell me where is _log_accuracy? image

xingyizhou commented 3 years ago

Hi, Thank you for raising up the issue! This is due to a detection2 version inconsistency. It has been fixed in the most recent commit.

xingyizhou commented 3 years ago

For training on custom datasets, please refer to the detectron2 tutorial here.

DatDoc commented 3 years ago

Hi @xingyizhou, Centernet2 hasn't been added to the model_zoo API so the model can just be trained using train_net.py? Amazing work btw

sh7jacobi commented 3 years ago

Hi. How can I set number of classes ? Is it enough to write in config file: " CENTERNET: NUM_CLASSES: 12 " ?

qdd1234 commented 3 years ago

yes，but i mention that you should change the class name at the same time

TfeiSong commented 3 years ago

yes，but i mention that you should change the class name at the same time

How do I change the class name in the code?