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=
|
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 = [
|
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:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6615790efd627772917205db291f51cd392528a157ecbb2ecaeec3bff8eb6de2
|
3 |
+
size 24636301
|