Spaces:

Chaerin5
/

FoundHand

Running on Zero

Chaerin5 commited on Dec 28, 2024

Commit

3a6a539

1 Parent(s): 4395911

enable zerogpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -338,7 +338,7 @@ def get_ref_anno(ref):
             Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5], inplace=True),
         ]
     )
-    image = image_transform(img) #.to(device)
     kpts_valid = check_keypoints_validity(keypts, opts.image_size)
     heatmaps = torch.tensor(
         keypoint_heatmap(

             Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5], inplace=True),
         ]
     )
+    image = image_transform(img)
     kpts_valid = check_keypoints_validity(keypts, opts.image_size)
     heatmaps = torch.tensor(
         keypoint_heatmap(