From ace2911acbc30e8f98fb0aa9686a3dc896442ac1 Mon Sep 17 00:00:00 2001
From: hzwer <598460606@163.com>
Date: Thu, 17 Jun 2021 18:02:06 +0800
Subject: [PATCH 1/3] Update README.md

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 9844bd0..dd20456 100644
--- a/README.md
+++ b/README.md
@@ -35,7 +35,7 @@ cd arXiv2020-RIFE
 pip3 install -r requirements.txt
 ```
 
-* Download the pretrained **HD** models from [here](https://drive.google.com/file/d/10-2AaFUyX-c7yCfubsxF2NTvM7DgvS8l/view?usp=sharing). (百度网盘链接:https://pan.baidu.com/s/1cJ7-dPuwR8THPUGWb207ZQ 密码:aa0w，把压缩包解开后放在 train_log/\*)
+* Download the pretrained **HD** models from [here](https://drive.google.com/file/d/1APIzVeI-4ZZCEuIRE1m6WYfSCaOsi_7_/view?usp=sharing). (百度网盘链接:https://pan.baidu.com/share/init?surl=u6Q7-i4Hu4Vx9_5BJibPPA 密码:hfk3，把压缩包解开后放在 train_log/\*)
 
 * Unzip and move the pretrained parameters to train_log/\*
 

From b7b503dc699973d703869a1ba70b7a6f9d480398 Mon Sep 17 00:00:00 2001
From: hzwer <598460606@163.com>
Date: Wed, 23 Jun 2021 12:19:04 +0800
Subject: [PATCH 2/3] Update inference_video.py

---
 inference_video.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/inference_video.py b/inference_video.py
index fd98e27..1befced 100644
--- a/inference_video.py
+++ b/inference_video.py
@@ -213,7 +213,7 @@ while True:
     I1 = pad_image(I1)
     I0_small = F.interpolate(I0, (32, 32), mode='bilinear', align_corners=False)
     I1_small = F.interpolate(I1, (32, 32), mode='bilinear', align_corners=False)
-    ssim = ssim_matlab(I0_small, I1_small)
+    ssim = ssim_matlab(I0_small[:, :3], I1_small[:, :3])
 
     if ssim > 0.995:
         if skip_frame % 100 == 0:

From 856d550d66e73c6f49ca59c249f3ddefd6cf7991 Mon Sep 17 00:00:00 2001
From: hzwer <598460606@163.com>
Date: Mon, 28 Jun 2021 15:33:21 +0800
Subject: [PATCH 3/3] Update RIFE.py

---
 model/RIFE.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/model/RIFE.py b/model/RIFE.py
index 9eaf5eb..22a1db1 100644
--- a/model/RIFE.py
+++ b/model/RIFE.py
@@ -183,7 +183,7 @@ class Model:
         else:
             return pred
 
-    def inference(self, img0, img1):
+    def inference(self, img0, img1, scale=None):
         imgs = torch.cat((img0, img1), 1)
         flow, _ = self.flownet(torch.cat((img0, img1), 1))
         return self.predict(imgs, flow, training=False)