Mirrors/ECCV2022-RIFE
1
0
Fork 0
mirror of https://github.com/hzwer/ECCV2022-RIFE.git synced 2026-02-24 04:19:41 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update train.py

Browse Source
This commit is contained in:
Zhewei Huang
2022-01-06 17:54:52 +08:00
committed by GitHub
parent 8eb9503046
commit 3ed182299c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@@ -119,7 +119,7 @@ def evaluate(model, val_data, nr_eval, local_rank, writer_val):
if i == 0 and local_rank == 0:
for j in range(10):
imgs = np.concatenate((merged_img[j], pred[j], gt[j]), 1)[:, :, ::-1]
..add_image(str(j) + '/img', imgs.copy(), nr_eval, dataformats='HWC')
writer_val.add_image(str(j) + '/img', imgs.copy(), nr_eval, dataformats='HWC')
writer_val.add_image(str(j) + '/flow', flow2rgb(flow0[j][:, :, ::-1]), nr_eval, dataformats='HWC')
eval_time_interval = time.time() - time_stamp