Mirrors/ECCV2022-RIFE
1
0
Fork 0
mirror of https://github.com/hzwer/ECCV2022-RIFE.git synced 2025-12-16 08:27:45 +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)
if not success2:
break
IMAGE1 = IMAGE1[:, :, ::-1].copy()
IMAGE2 = IMAGE2[:, :, ::-1].copy()
gt = gt[:, :, ::-1].copy()
else:
success1, gt = Reader.read()
success2, frame = Reader.read()

4
benchmark/HD_multi.py
View File

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