Mirrors/ECCV2022-RIFE
1
0
Fork 0
mirror of https://github.com/hzwer/ECCV2022-RIFE.git synced 2025-12-16 08:27:45 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update IFNet_HDv2.py

Browse Source
This commit is contained in:
hzwer
2021-02-08 22:03:52 +08:00
committed by GitHub
parent 7907d63a97
commit 9b0ebd20ed
1 changed files with 12 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
model/IFNet_HDv2.py
View File

@@ -25,19 +25,20 @@ class IFBlock(nn.Module):
def __init__(self, in_planes, scale=1, c=64): def __init__(self, in_planes, scale=1, c=64):
super(IFBlock, self).__init__() super(IFBlock, self).__init__()
self.scale = scale self.scale = scale
self.conv0 = conv(in_planes, c, 5, 2, 2) self.conv0 = nn.Sequential(
conv(in_planes, c, 3, 2, 1),
conv(c, 2*c, 3, 2, 1),
)
self.convblock = nn.Sequential( self.convblock = nn.Sequential(
conv(c, c), conv(2*c, 2*c),
conv(c, c), conv(2*c, 2*c),
conv(c, c), conv(2*c, 2*c),
conv(c, c), conv(2*c, 2*c),
conv(c, c), conv(2*c, 2*c),
conv(c, c), conv(2*c, 2*c),
conv(c, c),
conv(c, c),
) )
self.conv1 = nn.Conv2d(c, 4, 3, 1, 1) self.conv1 = nn.ConvTranspose2d(2*c, 4, 4, 2, 1)
def forward(self, x): def forward(self, x):
if self.scale != 1: if self.scale != 1:
x = F.interpolate(x, scale_factor=1. / self.scale, mode="bilinear", x = F.interpolate(x, scale_factor=1. / self.scale, mode="bilinear",