测试centernet的demo命令:
python demo.py ctdet --demo ../images/ --load_model ../models/ctdet_coco_dla_2x.pth
如果你也有这个问题,千万别一遍又一遍的装cuda,网上都说什么版本不匹配,no!
报错:Fix size testing. training chunk_sizes: [32] The output will be saved to /home/sys507/CenterNet-master/src/lib/../../exp/ctdet/default heads {'hm': 1, 'wh': 2, 'reg': 2} Creating model... loaded ../models/ctdet_coco_dla_2x.pth, epoch 230 Skip loading parameter hm.2.weight, required shapetorch.Size([1, 256, 1, 1]), loaded shapetorch.Size([80, 256, 1, 1]). If you see this, your model does not fully load the pre-trained weight. Please make sure you have correctly specified --arch xxx or set the correct --num_classes for your own dataset. Skip loading parameter hm.2.bias, required shapetorch.Size([1]), loaded shapetorch.Size(