训练自己的数据集 about centernet HOT 7 CLOSED

@jianghe9 Hi, you can format your own dataset in the same format as coco.

from centernet.

@Duankaiwen Can we load the pretrained CenterNet-104 model params to train our own data, then how to set the learning rate? Thx.

from centernet.

@HRanWang Hi, I think you can use the default learning rate at the beginning，if the loss were very unstable, you could reduce the learning rate.

from centernet.

@Duankaiwen Thanks! Then how to change the input_size (511 , 511) and output_sizes (128 , 128), how to design it?

from centernet.

@HRanWang modify the “input_size” and "out_sizes" in the config/xxx.json. Ceiling(input_size/out_sizes)=4

from centernet.

你训练完了吗？花了多长时间多少卡？

from centernet.

@HRanWang Can you tell me how to use the pretrained model to retrain my own data?

from centernet.