MonsterMMORPG commited on
Commit
4bbe4c1
·
verified ·
1 Parent(s): bd8f1a7

Upload 9 files

Browse files
train_log/IFNet_HDv3.py CHANGED
@@ -96,7 +96,7 @@ class IFNet(nn.Module):
96
  self.block1 = IFBlock(8+4+8+8, c=128)
97
  self.block2 = IFBlock(8+4+8+8, c=96)
98
  self.block3 = IFBlock(8+4+8+8, c=64)
99
- self.block4 = IFBlock(8+4+8+8, c=24)
100
  self.encode = Head()
101
 
102
  # not used during inference
 
96
  self.block1 = IFBlock(8+4+8+8, c=128)
97
  self.block2 = IFBlock(8+4+8+8, c=96)
98
  self.block3 = IFBlock(8+4+8+8, c=64)
99
+ self.block4 = IFBlock(8+4+8+8, c=32)
100
  self.encode = Head()
101
 
102
  # not used during inference
train_log/RIFE_HDv3.py CHANGED
@@ -55,7 +55,7 @@ class Model:
55
 
56
  def inference(self, img0, img1, timestep=0.5, scale=1.0):
57
  imgs = torch.cat((img0, img1), 1)
58
- scale_list = [32/scale, 16/scale, 8/scale, 4/scale, 1/scale]
59
  flow, mask, merged = self.flownet(imgs, timestep, scale_list)
60
  return merged[-1]
61
 
 
55
 
56
  def inference(self, img0, img1, timestep=0.5, scale=1.0):
57
  imgs = torch.cat((img0, img1), 1)
58
+ scale_list = [16/scale, 8/scale, 4/scale, 2/scale, 1/scale]
59
  flow, mask, merged = self.flownet(imgs, timestep, scale_list)
60
  return merged[-1]
61
 
train_log/__pycache__/IFNet_HDv3.cpython-310.pyc CHANGED
Binary files a/train_log/__pycache__/IFNet_HDv3.cpython-310.pyc and b/train_log/__pycache__/IFNet_HDv3.cpython-310.pyc differ
 
train_log/__pycache__/RIFE_HDv3.cpython-310.pyc CHANGED
Binary files a/train_log/__pycache__/RIFE_HDv3.cpython-310.pyc and b/train_log/__pycache__/RIFE_HDv3.cpython-310.pyc differ
 
train_log/__pycache__/RIFE_HDv3.cpython-311.pyc CHANGED
Binary files a/train_log/__pycache__/RIFE_HDv3.cpython-311.pyc and b/train_log/__pycache__/RIFE_HDv3.cpython-311.pyc differ
 
train_log/flownet.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:81cdba223fe72a120130cc8552e5d2ecac824259d406f0c15323b3decf96b8b1
3
- size 24467405
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6615790efd627772917205db291f51cd392528a157ecbb2ecaeec3bff8eb6de2
3
+ size 24636301