search

xingyizhou / CenterNet

Object detection, 3D detection, and pose estimation using center point detection:
MIT License
7.25k stars 1.92k forks source link

cannot run demo.py normally #202

Open yangyuya opened 5 years ago

yangyuya commented 5 years ago

when I run demo.py: python demo.py ctdet --demo ../CenterNet-1-master/images/17790319373_bd19b24cfc_k.jpg --load_model ../CenterNet-1-master/models/ctdet_coco_dla_2x.pth

it shows:

Fix size testing. training chunk_sizes: [1] The output will be saved to /home/mi6/目标检测/CenterNet-1-master/src/lib/../../exp/ctdet/default heads {'hm': 80, 'wh': 2, 'reg': 2} Creating model... loaded /home/mi6/目标检测/CenterNet-1-master/models/ctdet_coco_dla_2x.pth, epoch 230

then it doesn't work and doesn't show any error . I check it and find that it stop when run ret = detector.run(image_name) . I have no idea . How can I solve it?

xingyizhou commented 5 years ago

Can you check in which line inside detector.run does it stop? If it is in cv2.waitKey(), it might be that everything goes well and the images display in your other screen ...