Upload 9 files

train_log/IFNet_HDv3.py

@@ -27,7 +27,7 @@ class Head(nn.Module):
         self.cnn0 = nn.Conv2d(3, 16, 3, 2, 1)
         self.cnn1 = nn.Conv2d(16, 16, 3, 1, 1)
         self.cnn2 = nn.Conv2d(16, 16, 3, 1, 1)
-        self.cnn3 = nn.ConvTranspose2d(16, 16, 4, 2, 1)
+        self.cnn3 = nn.ConvTranspose2d(16, 4, 4, 2, 1)
         self.relu = nn.LeakyReLU(0.2, True)
 
     def forward(self, x, feat=False):
@@ -92,18 +92,18 @@ class IFBlock(nn.Module):
 class IFNet(nn.Module):
     def __init__(self):
         super(IFNet, self).__init__()
-        self.block0 = IFBlock(7+32, c=192)
-        self.block1 = IFBlock(8+4+8+32, c=128)
-        self.block2 = IFBlock(8+4+8+32, c=96)
-        self.block3 = IFBlock(8+4+8+32, c=64)
-        self.block4 = IFBlock(8+4+8+32, c=32)
+        self.block0 = IFBlock(7+8, c=192)
+        self.block1 = IFBlock(8+4+8+8, c=128)
+        self.block2 = IFBlock(8+4+8+8, c=96)
+        self.block3 = IFBlock(8+4+8+8, c=64)
+        self.block4 = IFBlock(8+4+8+8, c=24)
         self.encode = Head()
 
         # not used during inference
         '''
-        self.teacher = IFBlock(8+4+8+3+32, c=64)
+        self.teacher = IFBlock(8+4+8+3+8, c=64)
         self.caltime = nn.Sequential(
-            nn.Conv2d(32+9, 8, 3, 2, 1),
+            nn.Conv2d(16+9, 8, 3, 2, 1),
             nn.LeakyReLU(0.2, True),
             nn.Conv2d(32, 64, 3, 2, 1),
             nn.LeakyReLU(0.2, True),

train_log/RIFE_HDv3.py

@@ -55,7 +55,7 @@ class Model:
 
     def inference(self, img0, img1, timestep=0.5, scale=1.0):
         imgs = torch.cat((img0, img1), 1)
-        scale_list = [16/scale, 8/scale, 4/scale, 2/scale, 1/scale]
+        scale_list = [32/scale, 16/scale, 8/scale, 4/scale, 1/scale]
         flow, mask, merged = self.flownet(imgs, timestep, scale_list)
         return merged[-1]
     

train_log/flownet.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4cc518e172156ad6207b9c7a43364f518832d83a4325d484240493a9e2980537
-size 24925133
+oid sha256:81cdba223fe72a120130cc8552e5d2ecac824259d406f0c15323b3decf96b8b1
+size 24467405