CenterNet
CenterNet copied to clipboard
Is possible to run ModelZoo models without GPU?
Hi everyone, I followed the correct steps to run the demo. However, I got the following error: python demo.py multi_pose --demo 3.mp4 --load_model ../models/multi_pose_dla_3x.pth
Fix size testing.
training chunk_sizes: [32]
The output will be saved to /home/jrobador/CIII/CenterNet/CenterNet/src/lib/../../exp/multi_pose/default
heads {'hm': 1, 'wh': 2, 'hps': 34, 'reg': 2, 'hm_hp': 17, 'hp_offset': 2}
Creating model...
loaded ../models/multi_pose_dla_3x.pth, epoch 320
Traceback (most recent call last):
File "demo.py", line 56, in
I also tried using a tutorial that i founded on internet, (https://programmerclick.com/article/43101758055/) and i got the following error:
python demo.py multi_pose --demo 3.mp4 --load_model multi_pose_dla_3x.pth
Fix size testing.
training chunk_sizes: [32]
The output will be saved to /home/jrobador/CIII/CenterNet/CenterNet/src/lib/../../exp/multi_pose/default
heads {'hm': 1, 'wh': 2, 'hps': 34, 'reg': 2, 'hm_hp': 17, 'hp_offset': 2}
Creating model...
Downloading: "http://dl.yf.io/dla/models/imagenet/dla34-ba72cf86.pth" to /home/jrobador/.cache/torch/checkpoints/dla34-ba72cf86.pth
100.0%
Traceback (most recent call last):
File "demo.py", line 56, in
You have to go to the 'CenterNet/src/lib/detectors/base_detector.py' file and replace if opt.gpus[0] >= 0: opt.device = torch.device('cuda') else: opt.device = torch.device('cpu')
with opt.device = torch.device('cpu')
only. Also delete (or uncomment) any line that is torch.cuda.synchronize()
in 'CenterNet/src/lib/detectors/base_detector.py', 'CenterNet/src/lib/detectors/ctdet.py', 'CenterNet/src/lib/detectors/exdet.py', 'CenterNet/src/lib/detectors/ddd.py', 'CenterNet/src/lib/detectors/multi_pose.py'
I might be missing something but basically just delete any line which contain .cuda