Spaces:
Running
on
Zero
Running
on
Zero
enable zerogpu
Browse files
app.py
CHANGED
@@ -339,6 +339,8 @@ def get_ref_anno(ref):
|
|
339 |
]
|
340 |
)
|
341 |
image = image_transform(img)
|
|
|
|
|
342 |
kpts_valid = check_keypoints_validity(keypts, opts.image_size)
|
343 |
heatmaps = torch.tensor(
|
344 |
keypoint_heatmap(
|
@@ -358,7 +360,7 @@ def get_ref_anno(ref):
|
|
358 |
# device=device,
|
359 |
).unsqueeze(0)[None, ...]
|
360 |
image, latent = make_ref_cond(
|
361 |
-
image
|
362 |
# keypts,
|
363 |
# hand_mask,
|
364 |
# device=device,
|
|
|
339 |
]
|
340 |
)
|
341 |
image = image_transform(img)
|
342 |
+
image = image.numpy()
|
343 |
+
image = torch.tensor(image)
|
344 |
kpts_valid = check_keypoints_validity(keypts, opts.image_size)
|
345 |
heatmaps = torch.tensor(
|
346 |
keypoint_heatmap(
|
|
|
360 |
# device=device,
|
361 |
).unsqueeze(0)[None, ...]
|
362 |
image, latent = make_ref_cond(
|
363 |
+
image,
|
364 |
# keypts,
|
365 |
# hand_mask,
|
366 |
# device=device,
|