Mirrors/ECCV2022-RIFE
1
0
Fork 0
mirror of https://github.com/hzwer/ECCV2022-RIFE.git synced 2025-12-16 16:37:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

bgr to rgb

Browse Source
This commit is contained in:
hzwer
2021-03-06 08:55:50 +08:00
parent 9ec0ae18c6
commit 29e6801120
2 changed files with 1 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
benchmark/HD.py
View File

@@ -51,9 +51,6 @@ for data in name_list:
IMAGE2, success2 = Reader.read(index + 2) IMAGE2, success2 = Reader.read(index + 2)
if not success2: if not success2:
break break
IMAGE1 = IMAGE1[:, :, ::-1].copy()
IMAGE2 = IMAGE2[:, :, ::-1].copy()
gt = gt[:, :, ::-1].copy()
else: else:
success1, gt = Reader.read() success1, gt = Reader.read()
success2, frame = Reader.read() success2, frame = Reader.read()

4
benchmark/HD_multi.py
View File

@@ -65,9 +65,7 @@ for data in name_list:
break break
for i in range(1, 8): for i in range(1, 8):
tmp, _ = Reader.read(index + i) tmp, _ = Reader.read(index + i)
gt.append(tmp[:, :, ::-1].copy()) gt.append(tmp)
IMAGE1 = IMAGE1[:, :, ::-1].copy()
IMAGE2 = IMAGE2[:, :, ::-1].copy()
else: else:
print('Not Implement') print('Not Implement')
I0 = torch.from_numpy(np.transpose(IMAGE1, (2,0,1)).astype("float32") / 255.).cuda().unsqueeze(0) I0 = torch.from_numpy(np.transpose(IMAGE1, (2,0,1)).astype("float32") / 255.).cuda().unsqueeze(0)