HRNet-landmarks
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?
maybe I am stuck here... inside utils_inference.py file
img_crop = torch.tensor(img_crop, dtype=torch.float32).unsqueeze(0).cuda()
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