From dccd8e53c668a532792d46302caec7419de6e2be Mon Sep 17 00:00:00 2001 From: sniklaus Date: Fri, 24 Nov 2017 14:51:29 -0800 Subject: no message --- README.md | 1 + run.py | 16 +++++++++++----- src/SeparableConvolution_kernel.o | Bin 10984 -> 0 bytes 3 files changed, 12 insertions(+), 5 deletions(-) delete mode 100644 src/SeparableConvolution_kernel.o diff --git a/README.md b/README.md index 11dc862..754e6f8 100644 --- a/README.md +++ b/README.md @@ -4,6 +4,7 @@ This is a reference implementation of Video Frame Interpolation via Adaptive Sep Paper For the Torch version of this work, please see: https://github.com/sniklaus/torch-sepconv +For a third-party fork with video support, consider: https://github.com/dagf2101/pytorch-sepconv ## setup To build the implementation and download the pre-trained networks, run `bash install.bash` and make sure that you configured the `CUDA_HOME` environment variable. After successfully completing this step, run `python run.py` to test it. Should you receive an error message regarding an invalid device function during execution, configure the utilized CUDA architecture within `install.bash` to something your graphics card supports. diff --git a/run.py b/run.py index a1f26a3..a7a7add 100644 --- a/run.py +++ b/run.py @@ -142,13 +142,19 @@ class Network(torch.nn.Module): variableDeconv5 = self.moduleDeconv5(variablePool5) variableUpsample5 = self.moduleUpsample5(variableDeconv5) - variableDeconv4 = self.moduleDeconv4(variableUpsample5 + variableConv5) + variableCombine = variableUpsample5 + variableConv5 + + variableDeconv4 = self.moduleDeconv4(variableCombine) variableUpsample4 = self.moduleUpsample4(variableDeconv4) - variableDeconv3 = self.moduleDeconv3(variableUpsample4 + variableConv4) + variableCombine = variableUpsample4 + variableConv4 + + variableDeconv3 = self.moduleDeconv3(variableCombine) variableUpsample3 = self.moduleUpsample3(variableDeconv3) - variableDeconv2 = self.moduleDeconv2(variableUpsample3 + variableConv3) + variableCombine = variableUpsample3 + variableConv3 + + variableDeconv2 = self.moduleDeconv2(variableCombine) variableUpsample2 = self.moduleUpsample2(variableDeconv2) variableCombine = variableUpsample2 + variableConv2 @@ -181,8 +187,8 @@ intPaddingLeft = int(math.floor(51 / 2.0)) intPaddingTop = int(math.floor(51 / 2.0)) intPaddingRight = int(math.floor(51 / 2.0)) intPaddingBottom = int(math.floor(51 / 2.0)) -modulePaddingInput = torch.nn.Module() -modulePaddingOutput = torch.nn.Module() +modulePaddingInput = torch.nn.Sequential() +modulePaddingOutput = torch.nn.Sequential() if True: intPaddingWidth = intPaddingLeft + intWidth + intPaddingRight diff --git a/src/SeparableConvolution_kernel.o b/src/SeparableConvolution_kernel.o deleted file mode 100644 index 396aba1..0000000 Binary files a/src/SeparableConvolution_kernel.o and /dev/null differ -- cgit v1.2.3-70-g09d2