I am working on a project which requires turning a picture of a human into a 3D - Model. I am facing an error while running a block of code. I can't find its solution anywhere. I apologize if it is a stupid question as I am new to this field.
The block of code I am having issues with is:
net = PoseEstimationWithMobileNet()
checkpoint = torch.load('checkpoint_iter_370000.pth', map_location='cpu')
load_state(net, checkpoint)
get_rect(net.cuda(), [image_path], 512)
Error:
NameError: name 'PoseEstimationWithMobileNet' is not defined
Does anyone have an idea of where "PoseEstimationWithMobileNet" comes from?