Doubt result using resnet-101 pascal

xingyizhou / CenterNet

Object detection, 3D detection, and pose estimation using center point detection:

MIT License

7.24k stars 1.92k forks source link

Doubt result using resnet-101 pascal #352

Open rafikg opened 4 years ago

rafikg commented 4 years ago

Hello, I want to try the accuracy of resnet-101 pretrained on pascal (512*512)by running this script python demo.py ctdet --demo webcam --load_model ../models/ctdet_pascal_resdcn101_512.pth I tried with different videos and the result was nothing (The model does not detect any object at all)!! Is it normal!

rafikg commented 4 years ago

It is resolved, I forget to specify the architecture but the result is completely poor. The model generated a lot of bboxes!!

xingyizhou commented 4 years ago

It is because you haven't loaded the weights of the last layer. demo.py only supports output 80 classes, while your weight only has 20. You will need to modify the update head code in opt.py to support 20 classes.