yerang commited on
Commit
229976e
·
verified ·
1 Parent(s): b55b510

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