Spaces:
Sleeping
Sleeping
Update stf/stf-api-alternative/src/stf_alternative/dataset.py
Browse files
stf/stf-api-alternative/src/stf_alternative/dataset.py
CHANGED
@@ -53,6 +53,9 @@ class LipGanImage(Dataset):
|
|
53 |
sidx = frame_id(gt_fname)
|
54 |
img_gt = cv2.imread(str(img_name), cv2.IMREAD_UNCHANGED)
|
55 |
|
|
|
|
|
|
|
56 |
masked = img_gt[:, :, :3].copy()
|
57 |
img_ip = masked * 2.0 / 255.0 - 1.0
|
58 |
|
|
|
53 |
sidx = frame_id(gt_fname)
|
54 |
img_gt = cv2.imread(str(img_name), cv2.IMREAD_UNCHANGED)
|
55 |
|
56 |
+
print('img_name',str(img_name))
|
57 |
+
print('img_gt',img_gt)
|
58 |
+
|
59 |
masked = img_gt[:, :, :3].copy()
|
60 |
img_ip = masked * 2.0 / 255.0 - 1.0
|
61 |
|