search

xingyizhou / CenterTrack

Simultaneous object detection and tracking using center points.
MIT License
2.37k stars 526 forks source link

Train CenterNet with this code #186

Open zhangxgu opened 3 years ago

zhangxgu commented 3 years ago

Hi! It's a nice work!! I wonder if I can train a CenterNet for image detection on coco? I train the model with ''' python main.py ctdet --exp_id coco_dla_1x --batch_size 128 --master_batch 9 --lr 5e-4 --gpus 0,1,2,3,4,5,6,7 --num_workers 16 ''' And get loss like this: ep70 |Tot: 0:17:02 |ETA: 0:00:02 |tot 3.8847 |hm 2.9095 |wh 7.3402 |reg 0.2412 |

Since the there are many differences in opts.py of CenterTrack and CenterNet, I wonder if I can train a CenterNet with this code. Thanks!

xingyizhou commented 3 years ago

Hi, Sorry for the delayed reply. Yes this code supports the full functionality of CenterNet.