HRNet-landmarks icon indicating copy to clipboard operation
HRNet-landmarks copied to clipboard

hey i am trying to run inference on cpu coz my gpu is not cuda enabled. so can anybody help me to run this code?

Open Onkar-G opened this issue 5 years ago • 1 comments

maybe I am stuck here... inside utils_inference.py file

img_crop = torch.tensor(img_crop, dtype=torch.float32).unsqueeze(0).cuda()

Onkar-G avatar Nov 04 '20 22:11 Onkar-G

Delete cuda() img_crop = torch.tensor(img_crop, dtype=torch.float32).unsqueeze(0) You also have to change the device arguments to "CPU" when you call get_model_by_name

AlexandreLamb avatar Sep 16 '22 10:09 AlexandreLamb