no message

sniklaus · Nov 19, 2018 · 11ad5c5 · 11ad5c5
1 parent 471bd7a
commit 11ad5c5
Show file tree

Hide file tree

Showing 284 changed files with 14 additions and 29 deletions.
diff --git a/README.md b/README.md
@@ -6,6 +6,11 @@ This is a personal reimplementation of Optical Flow Estimation using a Spatial P
 For the original Torch version of this work, please see: https://github.com/anuragranj/spynet
 <br />
 Another optical flow implementation from me: https://github.com/sniklaus/pytorch-pwc
+<br />
+And another optical flow implementation from me: https://github.com/sniklaus/pytorch-unflow
+
+## setup
+To download the pre-trained models, run `bash download.bash`. These originate from the original authors, I just converted them to PyTorch.
 
 ## usage
 To run it on your own pair of images, use the following command. You can choose between various models, please make sure to see their paper / the code for more details.

diff --git a/download.bash b/download.bash
@@ -0,0 +1,7 @@
+#!/bin/bash
+
+wget --timestamping http://content.sniklaus.com/github/pytorch-spynet/network-chairs-clean.pytorch
+wget --timestamping http://content.sniklaus.com/github/pytorch-spynet/network-chairs-final.pytorch
+wget --timestamping http://content.sniklaus.com/github/pytorch-spynet/network-kitti-final.pytorch
+wget --timestamping http://content.sniklaus.com/github/pytorch-spynet/network-sintel-clean.pytorch
+wget --timestamping http://content.sniklaus.com/github/pytorch-spynet/network-sintel-final.pytorch
diff --git a/models/README.md b/models/README.md
diff --git a/models/modelL1_3-1-bias.t7 b/models/modelL1_3-1-bias.t7
diff --git a/models/modelL1_3-1-weight.t7 b/models/modelL1_3-1-weight.t7
diff --git a/models/modelL1_3-2-bias.t7 b/models/modelL1_3-2-bias.t7
diff --git a/models/modelL1_3-2-weight.t7 b/models/modelL1_3-2-weight.t7
diff --git a/models/modelL1_3-3-bias.t7 b/models/modelL1_3-3-bias.t7
diff --git a/models/modelL1_3-3-weight.t7 b/models/modelL1_3-3-weight.t7
diff --git a/models/modelL1_3-4-bias.t7 b/models/modelL1_3-4-bias.t7
diff --git a/models/modelL1_3-4-weight.t7 b/models/modelL1_3-4-weight.t7
diff --git a/models/modelL1_3-5-bias.t7 b/models/modelL1_3-5-bias.t7
diff --git a/models/modelL1_3-5-weight.t7 b/models/modelL1_3-5-weight.t7
diff --git a/models/modelL1_4-1-bias.t7 b/models/modelL1_4-1-bias.t7
diff --git a/models/modelL1_4-1-weight.t7 b/models/modelL1_4-1-weight.t7
diff --git a/models/modelL1_4-2-bias.t7 b/models/modelL1_4-2-bias.t7
diff --git a/models/modelL1_4-2-weight.t7 b/models/modelL1_4-2-weight.t7
diff --git a/models/modelL1_4-3-bias.t7 b/models/modelL1_4-3-bias.t7
diff --git a/models/modelL1_4-3-weight.t7 b/models/modelL1_4-3-weight.t7
diff --git a/models/modelL1_4-4-bias.t7 b/models/modelL1_4-4-bias.t7
diff --git a/models/modelL1_4-4-weight.t7 b/models/modelL1_4-4-weight.t7
diff --git a/models/modelL1_4-5-bias.t7 b/models/modelL1_4-5-bias.t7
diff --git a/models/modelL1_4-5-weight.t7 b/models/modelL1_4-5-weight.t7
diff --git a/models/modelL1_C-1-bias.t7 b/models/modelL1_C-1-bias.t7
diff --git a/models/modelL1_C-1-weight.t7 b/models/modelL1_C-1-weight.t7
diff --git a/models/modelL1_C-2-bias.t7 b/models/modelL1_C-2-bias.t7
diff --git a/models/modelL1_C-2-weight.t7 b/models/modelL1_C-2-weight.t7
diff --git a/models/modelL1_C-3-bias.t7 b/models/modelL1_C-3-bias.t7
diff --git a/models/modelL1_C-3-weight.t7 b/models/modelL1_C-3-weight.t7
diff --git a/models/modelL1_C-4-bias.t7 b/models/modelL1_C-4-bias.t7
diff --git a/models/modelL1_C-4-weight.t7 b/models/modelL1_C-4-weight.t7
diff --git a/models/modelL1_C-5-bias.t7 b/models/modelL1_C-5-bias.t7
diff --git a/models/modelL1_C-5-weight.t7 b/models/modelL1_C-5-weight.t7
diff --git a/models/modelL1_F-1-bias.t7 b/models/modelL1_F-1-bias.t7
diff --git a/models/modelL1_F-1-weight.t7 b/models/modelL1_F-1-weight.t7
diff --git a/models/modelL1_F-2-bias.t7 b/models/modelL1_F-2-bias.t7
diff --git a/models/modelL1_F-2-weight.t7 b/models/modelL1_F-2-weight.t7
diff --git a/models/modelL1_F-3-bias.t7 b/models/modelL1_F-3-bias.t7
diff --git a/models/modelL1_F-3-weight.t7 b/models/modelL1_F-3-weight.t7
diff --git a/models/modelL1_F-4-bias.t7 b/models/modelL1_F-4-bias.t7
diff --git a/models/modelL1_F-4-weight.t7 b/models/modelL1_F-4-weight.t7
diff --git a/models/modelL1_F-5-bias.t7 b/models/modelL1_F-5-bias.t7
diff --git a/models/modelL1_F-5-weight.t7 b/models/modelL1_F-5-weight.t7
diff --git a/models/modelL1_K-1-bias.t7 b/models/modelL1_K-1-bias.t7
diff --git a/models/modelL1_K-1-weight.t7 b/models/modelL1_K-1-weight.t7
diff --git a/models/modelL1_K-2-bias.t7 b/models/modelL1_K-2-bias.t7
diff --git a/models/modelL1_K-2-weight.t7 b/models/modelL1_K-2-weight.t7
diff --git a/models/modelL1_K-3-bias.t7 b/models/modelL1_K-3-bias.t7
diff --git a/models/modelL1_K-3-weight.t7 b/models/modelL1_K-3-weight.t7
diff --git a/models/modelL1_K-4-bias.t7 b/models/modelL1_K-4-bias.t7
diff --git a/models/modelL1_K-4-weight.t7 b/models/modelL1_K-4-weight.t7
diff --git a/models/modelL1_K-5-bias.t7 b/models/modelL1_K-5-bias.t7
diff --git a/models/modelL1_K-5-weight.t7 b/models/modelL1_K-5-weight.t7
diff --git a/models/modelL2_3-1-bias.t7 b/models/modelL2_3-1-bias.t7
diff --git a/models/modelL2_3-1-weight.t7 b/models/modelL2_3-1-weight.t7
diff --git a/models/modelL2_3-2-bias.t7 b/models/modelL2_3-2-bias.t7
diff --git a/models/modelL2_3-2-weight.t7 b/models/modelL2_3-2-weight.t7
diff --git a/models/modelL2_3-3-bias.t7 b/models/modelL2_3-3-bias.t7
diff --git a/models/modelL2_3-3-weight.t7 b/models/modelL2_3-3-weight.t7
diff --git a/models/modelL2_3-4-bias.t7 b/models/modelL2_3-4-bias.t7
diff --git a/models/modelL2_3-4-weight.t7 b/models/modelL2_3-4-weight.t7
diff --git a/models/modelL2_3-5-bias.t7 b/models/modelL2_3-5-bias.t7
diff --git a/models/modelL2_3-5-weight.t7 b/models/modelL2_3-5-weight.t7
diff --git a/models/modelL2_4-1-bias.t7 b/models/modelL2_4-1-bias.t7
diff --git a/models/modelL2_4-1-weight.t7 b/models/modelL2_4-1-weight.t7
diff --git a/models/modelL2_4-2-bias.t7 b/models/modelL2_4-2-bias.t7
diff --git a/models/modelL2_4-2-weight.t7 b/models/modelL2_4-2-weight.t7
diff --git a/models/modelL2_4-3-bias.t7 b/models/modelL2_4-3-bias.t7
diff --git a/models/modelL2_4-3-weight.t7 b/models/modelL2_4-3-weight.t7
diff --git a/models/modelL2_4-4-bias.t7 b/models/modelL2_4-4-bias.t7
diff --git a/models/modelL2_4-4-weight.t7 b/models/modelL2_4-4-weight.t7
diff --git a/models/modelL2_4-5-bias.t7 b/models/modelL2_4-5-bias.t7
diff --git a/models/modelL2_4-5-weight.t7 b/models/modelL2_4-5-weight.t7
diff --git a/models/modelL2_C-1-bias.t7 b/models/modelL2_C-1-bias.t7
diff --git a/models/modelL2_C-1-weight.t7 b/models/modelL2_C-1-weight.t7
diff --git a/models/modelL2_C-2-bias.t7 b/models/modelL2_C-2-bias.t7
diff --git a/models/modelL2_C-2-weight.t7 b/models/modelL2_C-2-weight.t7
diff --git a/models/modelL2_C-3-bias.t7 b/models/modelL2_C-3-bias.t7
diff --git a/models/modelL2_C-3-weight.t7 b/models/modelL2_C-3-weight.t7
diff --git a/models/modelL2_C-4-bias.t7 b/models/modelL2_C-4-bias.t7
diff --git a/models/modelL2_C-4-weight.t7 b/models/modelL2_C-4-weight.t7
diff --git a/models/modelL2_C-5-bias.t7 b/models/modelL2_C-5-bias.t7
diff --git a/models/modelL2_C-5-weight.t7 b/models/modelL2_C-5-weight.t7
diff --git a/models/modelL2_F-1-bias.t7 b/models/modelL2_F-1-bias.t7
diff --git a/models/modelL2_F-1-weight.t7 b/models/modelL2_F-1-weight.t7
diff --git a/models/modelL2_F-2-bias.t7 b/models/modelL2_F-2-bias.t7
diff --git a/models/modelL2_F-2-weight.t7 b/models/modelL2_F-2-weight.t7
diff --git a/models/modelL2_F-3-bias.t7 b/models/modelL2_F-3-bias.t7
diff --git a/models/modelL2_F-3-weight.t7 b/models/modelL2_F-3-weight.t7
diff --git a/models/modelL2_F-4-bias.t7 b/models/modelL2_F-4-bias.t7
diff --git a/models/modelL2_F-4-weight.t7 b/models/modelL2_F-4-weight.t7
diff --git a/models/modelL2_F-5-bias.t7 b/models/modelL2_F-5-bias.t7
diff --git a/models/modelL2_F-5-weight.t7 b/models/modelL2_F-5-weight.t7
diff --git a/models/modelL2_K-1-bias.t7 b/models/modelL2_K-1-bias.t7
diff --git a/models/modelL2_K-1-weight.t7 b/models/modelL2_K-1-weight.t7
diff --git a/models/modelL2_K-2-bias.t7 b/models/modelL2_K-2-bias.t7
diff --git a/models/modelL2_K-2-weight.t7 b/models/modelL2_K-2-weight.t7
diff --git a/models/modelL2_K-3-bias.t7 b/models/modelL2_K-3-bias.t7
diff --git a/models/modelL2_K-3-weight.t7 b/models/modelL2_K-3-weight.t7
diff --git a/models/modelL2_K-4-bias.t7 b/models/modelL2_K-4-bias.t7
diff --git a/models/modelL2_K-4-weight.t7 b/models/modelL2_K-4-weight.t7
diff --git a/models/modelL2_K-5-bias.t7 b/models/modelL2_K-5-bias.t7
diff --git a/models/modelL2_K-5-weight.t7 b/models/modelL2_K-5-weight.t7
diff --git a/models/modelL3_3-1-bias.t7 b/models/modelL3_3-1-bias.t7
diff --git a/models/modelL3_3-1-weight.t7 b/models/modelL3_3-1-weight.t7
diff --git a/models/modelL3_3-2-bias.t7 b/models/modelL3_3-2-bias.t7
diff --git a/models/modelL3_3-2-weight.t7 b/models/modelL3_3-2-weight.t7
diff --git a/models/modelL3_3-3-bias.t7 b/models/modelL3_3-3-bias.t7
diff --git a/models/modelL3_3-3-weight.t7 b/models/modelL3_3-3-weight.t7
diff --git a/models/modelL3_3-4-bias.t7 b/models/modelL3_3-4-bias.t7
diff --git a/models/modelL3_3-4-weight.t7 b/models/modelL3_3-4-weight.t7
diff --git a/models/modelL3_3-5-bias.t7 b/models/modelL3_3-5-bias.t7
diff --git a/models/modelL3_3-5-weight.t7 b/models/modelL3_3-5-weight.t7
diff --git a/models/modelL3_4-1-bias.t7 b/models/modelL3_4-1-bias.t7
diff --git a/models/modelL3_4-1-weight.t7 b/models/modelL3_4-1-weight.t7
diff --git a/models/modelL3_4-2-bias.t7 b/models/modelL3_4-2-bias.t7
diff --git a/models/modelL3_4-2-weight.t7 b/models/modelL3_4-2-weight.t7
diff --git a/models/modelL3_4-3-bias.t7 b/models/modelL3_4-3-bias.t7
diff --git a/models/modelL3_4-3-weight.t7 b/models/modelL3_4-3-weight.t7
diff --git a/models/modelL3_4-4-bias.t7 b/models/modelL3_4-4-bias.t7
diff --git a/models/modelL3_4-4-weight.t7 b/models/modelL3_4-4-weight.t7
diff --git a/models/modelL3_4-5-bias.t7 b/models/modelL3_4-5-bias.t7
diff --git a/models/modelL3_4-5-weight.t7 b/models/modelL3_4-5-weight.t7
diff --git a/models/modelL3_C-1-bias.t7 b/models/modelL3_C-1-bias.t7
diff --git a/models/modelL3_C-1-weight.t7 b/models/modelL3_C-1-weight.t7
diff --git a/models/modelL3_C-2-bias.t7 b/models/modelL3_C-2-bias.t7
diff --git a/models/modelL3_C-2-weight.t7 b/models/modelL3_C-2-weight.t7
diff --git a/models/modelL3_C-3-bias.t7 b/models/modelL3_C-3-bias.t7
diff --git a/models/modelL3_C-3-weight.t7 b/models/modelL3_C-3-weight.t7
diff --git a/models/modelL3_C-4-bias.t7 b/models/modelL3_C-4-bias.t7
diff --git a/models/modelL3_C-4-weight.t7 b/models/modelL3_C-4-weight.t7
diff --git a/models/modelL3_C-5-bias.t7 b/models/modelL3_C-5-bias.t7
diff --git a/models/modelL3_C-5-weight.t7 b/models/modelL3_C-5-weight.t7
diff --git a/models/modelL3_F-1-bias.t7 b/models/modelL3_F-1-bias.t7
diff --git a/models/modelL3_F-1-weight.t7 b/models/modelL3_F-1-weight.t7
diff --git a/models/modelL3_F-2-bias.t7 b/models/modelL3_F-2-bias.t7
diff --git a/models/modelL3_F-2-weight.t7 b/models/modelL3_F-2-weight.t7
diff --git a/models/modelL3_F-3-bias.t7 b/models/modelL3_F-3-bias.t7
diff --git a/models/modelL3_F-3-weight.t7 b/models/modelL3_F-3-weight.t7
diff --git a/models/modelL3_F-4-bias.t7 b/models/modelL3_F-4-bias.t7
diff --git a/models/modelL3_F-4-weight.t7 b/models/modelL3_F-4-weight.t7
diff --git a/models/modelL3_F-5-bias.t7 b/models/modelL3_F-5-bias.t7
diff --git a/models/modelL3_F-5-weight.t7 b/models/modelL3_F-5-weight.t7
diff --git a/models/modelL3_K-1-bias.t7 b/models/modelL3_K-1-bias.t7
diff --git a/models/modelL3_K-1-weight.t7 b/models/modelL3_K-1-weight.t7
diff --git a/models/modelL3_K-2-bias.t7 b/models/modelL3_K-2-bias.t7
diff --git a/models/modelL3_K-2-weight.t7 b/models/modelL3_K-2-weight.t7
diff --git a/models/modelL3_K-3-bias.t7 b/models/modelL3_K-3-bias.t7
diff --git a/models/modelL3_K-3-weight.t7 b/models/modelL3_K-3-weight.t7
diff --git a/models/modelL3_K-4-bias.t7 b/models/modelL3_K-4-bias.t7
diff --git a/models/modelL3_K-4-weight.t7 b/models/modelL3_K-4-weight.t7
diff --git a/models/modelL3_K-5-bias.t7 b/models/modelL3_K-5-bias.t7
diff --git a/models/modelL3_K-5-weight.t7 b/models/modelL3_K-5-weight.t7
diff --git a/models/modelL4_3-1-bias.t7 b/models/modelL4_3-1-bias.t7
diff --git a/models/modelL4_3-1-weight.t7 b/models/modelL4_3-1-weight.t7
diff --git a/models/modelL4_3-2-bias.t7 b/models/modelL4_3-2-bias.t7
diff --git a/models/modelL4_3-2-weight.t7 b/models/modelL4_3-2-weight.t7
diff --git a/models/modelL4_3-3-bias.t7 b/models/modelL4_3-3-bias.t7
diff --git a/models/modelL4_3-3-weight.t7 b/models/modelL4_3-3-weight.t7
diff --git a/models/modelL4_3-4-bias.t7 b/models/modelL4_3-4-bias.t7
diff --git a/models/modelL4_3-4-weight.t7 b/models/modelL4_3-4-weight.t7
diff --git a/models/modelL4_3-5-bias.t7 b/models/modelL4_3-5-bias.t7
diff --git a/models/modelL4_3-5-weight.t7 b/models/modelL4_3-5-weight.t7
diff --git a/models/modelL4_4-1-bias.t7 b/models/modelL4_4-1-bias.t7
diff --git a/models/modelL4_4-1-weight.t7 b/models/modelL4_4-1-weight.t7
diff --git a/models/modelL4_4-2-bias.t7 b/models/modelL4_4-2-bias.t7
diff --git a/models/modelL4_4-2-weight.t7 b/models/modelL4_4-2-weight.t7
diff --git a/models/modelL4_4-3-bias.t7 b/models/modelL4_4-3-bias.t7
diff --git a/models/modelL4_4-3-weight.t7 b/models/modelL4_4-3-weight.t7
diff --git a/models/modelL4_4-4-bias.t7 b/models/modelL4_4-4-bias.t7
diff --git a/models/modelL4_4-4-weight.t7 b/models/modelL4_4-4-weight.t7
diff --git a/models/modelL4_4-5-bias.t7 b/models/modelL4_4-5-bias.t7
diff --git a/models/modelL4_4-5-weight.t7 b/models/modelL4_4-5-weight.t7
diff --git a/models/modelL4_C-1-bias.t7 b/models/modelL4_C-1-bias.t7
diff --git a/models/modelL4_C-1-weight.t7 b/models/modelL4_C-1-weight.t7
diff --git a/models/modelL4_C-2-bias.t7 b/models/modelL4_C-2-bias.t7
diff --git a/models/modelL4_C-2-weight.t7 b/models/modelL4_C-2-weight.t7
diff --git a/models/modelL4_C-3-bias.t7 b/models/modelL4_C-3-bias.t7
diff --git a/models/modelL4_C-3-weight.t7 b/models/modelL4_C-3-weight.t7
diff --git a/models/modelL4_C-4-bias.t7 b/models/modelL4_C-4-bias.t7
diff --git a/models/modelL4_C-4-weight.t7 b/models/modelL4_C-4-weight.t7
diff --git a/models/modelL4_C-5-bias.t7 b/models/modelL4_C-5-bias.t7
diff --git a/models/modelL4_C-5-weight.t7 b/models/modelL4_C-5-weight.t7
diff --git a/models/modelL4_F-1-bias.t7 b/models/modelL4_F-1-bias.t7
diff --git a/models/modelL4_F-1-weight.t7 b/models/modelL4_F-1-weight.t7
diff --git a/models/modelL4_F-2-bias.t7 b/models/modelL4_F-2-bias.t7
diff --git a/models/modelL4_F-2-weight.t7 b/models/modelL4_F-2-weight.t7
diff --git a/models/modelL4_F-3-bias.t7 b/models/modelL4_F-3-bias.t7
diff --git a/models/modelL4_F-3-weight.t7 b/models/modelL4_F-3-weight.t7
diff --git a/models/modelL4_F-4-bias.t7 b/models/modelL4_F-4-bias.t7
diff --git a/models/modelL4_F-4-weight.t7 b/models/modelL4_F-4-weight.t7
diff --git a/models/modelL4_F-5-bias.t7 b/models/modelL4_F-5-bias.t7
diff --git a/models/modelL4_F-5-weight.t7 b/models/modelL4_F-5-weight.t7
diff --git a/models/modelL4_K-1-bias.t7 b/models/modelL4_K-1-bias.t7
diff --git a/models/modelL4_K-1-weight.t7 b/models/modelL4_K-1-weight.t7
diff --git a/models/modelL4_K-2-bias.t7 b/models/modelL4_K-2-bias.t7
diff --git a/models/modelL4_K-2-weight.t7 b/models/modelL4_K-2-weight.t7
diff --git a/models/modelL4_K-3-bias.t7 b/models/modelL4_K-3-bias.t7
diff --git a/models/modelL4_K-3-weight.t7 b/models/modelL4_K-3-weight.t7
diff --git a/models/modelL4_K-4-bias.t7 b/models/modelL4_K-4-bias.t7
diff --git a/models/modelL4_K-4-weight.t7 b/models/modelL4_K-4-weight.t7
diff --git a/models/modelL4_K-5-bias.t7 b/models/modelL4_K-5-bias.t7
diff --git a/models/modelL4_K-5-weight.t7 b/models/modelL4_K-5-weight.t7
diff --git a/models/modelL5_3-1-bias.t7 b/models/modelL5_3-1-bias.t7
diff --git a/models/modelL5_3-1-weight.t7 b/models/modelL5_3-1-weight.t7
diff --git a/models/modelL5_3-2-bias.t7 b/models/modelL5_3-2-bias.t7
diff --git a/models/modelL5_3-2-weight.t7 b/models/modelL5_3-2-weight.t7
diff --git a/models/modelL5_3-3-bias.t7 b/models/modelL5_3-3-bias.t7
diff --git a/models/modelL5_3-3-weight.t7 b/models/modelL5_3-3-weight.t7
diff --git a/models/modelL5_3-4-bias.t7 b/models/modelL5_3-4-bias.t7
diff --git a/models/modelL5_3-4-weight.t7 b/models/modelL5_3-4-weight.t7
diff --git a/models/modelL5_3-5-bias.t7 b/models/modelL5_3-5-bias.t7
diff --git a/models/modelL5_3-5-weight.t7 b/models/modelL5_3-5-weight.t7
diff --git a/models/modelL5_4-1-bias.t7 b/models/modelL5_4-1-bias.t7
diff --git a/models/modelL5_4-1-weight.t7 b/models/modelL5_4-1-weight.t7
diff --git a/models/modelL5_4-2-bias.t7 b/models/modelL5_4-2-bias.t7
diff --git a/models/modelL5_4-2-weight.t7 b/models/modelL5_4-2-weight.t7
diff --git a/models/modelL5_4-3-bias.t7 b/models/modelL5_4-3-bias.t7
diff --git a/models/modelL5_4-3-weight.t7 b/models/modelL5_4-3-weight.t7
diff --git a/models/modelL5_4-4-bias.t7 b/models/modelL5_4-4-bias.t7
diff --git a/models/modelL5_4-4-weight.t7 b/models/modelL5_4-4-weight.t7
diff --git a/models/modelL5_4-5-bias.t7 b/models/modelL5_4-5-bias.t7
diff --git a/models/modelL5_4-5-weight.t7 b/models/modelL5_4-5-weight.t7
diff --git a/models/modelL5_C-1-bias.t7 b/models/modelL5_C-1-bias.t7
diff --git a/models/modelL5_C-1-weight.t7 b/models/modelL5_C-1-weight.t7
diff --git a/models/modelL5_C-2-bias.t7 b/models/modelL5_C-2-bias.t7
diff --git a/models/modelL5_C-2-weight.t7 b/models/modelL5_C-2-weight.t7
diff --git a/models/modelL5_C-3-bias.t7 b/models/modelL5_C-3-bias.t7
diff --git a/models/modelL5_C-3-weight.t7 b/models/modelL5_C-3-weight.t7
diff --git a/models/modelL5_C-4-bias.t7 b/models/modelL5_C-4-bias.t7
diff --git a/models/modelL5_C-4-weight.t7 b/models/modelL5_C-4-weight.t7
diff --git a/models/modelL5_C-5-bias.t7 b/models/modelL5_C-5-bias.t7
diff --git a/models/modelL5_C-5-weight.t7 b/models/modelL5_C-5-weight.t7
diff --git a/models/modelL5_F-1-bias.t7 b/models/modelL5_F-1-bias.t7
diff --git a/models/modelL5_F-1-weight.t7 b/models/modelL5_F-1-weight.t7
diff --git a/models/modelL5_F-2-bias.t7 b/models/modelL5_F-2-bias.t7
diff --git a/models/modelL5_F-2-weight.t7 b/models/modelL5_F-2-weight.t7
diff --git a/models/modelL5_F-3-bias.t7 b/models/modelL5_F-3-bias.t7
diff --git a/models/modelL5_F-3-weight.t7 b/models/modelL5_F-3-weight.t7
diff --git a/models/modelL5_F-4-bias.t7 b/models/modelL5_F-4-bias.t7
diff --git a/models/modelL5_F-4-weight.t7 b/models/modelL5_F-4-weight.t7
diff --git a/models/modelL5_F-5-bias.t7 b/models/modelL5_F-5-bias.t7
diff --git a/models/modelL5_F-5-weight.t7 b/models/modelL5_F-5-weight.t7
diff --git a/models/modelL5_K-1-bias.t7 b/models/modelL5_K-1-bias.t7
diff --git a/models/modelL5_K-1-weight.t7 b/models/modelL5_K-1-weight.t7
diff --git a/models/modelL5_K-2-bias.t7 b/models/modelL5_K-2-bias.t7
diff --git a/models/modelL5_K-2-weight.t7 b/models/modelL5_K-2-weight.t7
diff --git a/models/modelL5_K-3-bias.t7 b/models/modelL5_K-3-bias.t7
diff --git a/models/modelL5_K-3-weight.t7 b/models/modelL5_K-3-weight.t7
diff --git a/models/modelL5_K-4-bias.t7 b/models/modelL5_K-4-bias.t7
diff --git a/models/modelL5_K-4-weight.t7 b/models/modelL5_K-4-weight.t7
diff --git a/models/modelL5_K-5-bias.t7 b/models/modelL5_K-5-bias.t7
diff --git a/models/modelL5_K-5-weight.t7 b/models/modelL5_K-5-weight.t7
diff --git a/models/modelL6_C-1-bias.t7 b/models/modelL6_C-1-bias.t7
diff --git a/models/modelL6_C-1-weight.t7 b/models/modelL6_C-1-weight.t7
diff --git a/models/modelL6_C-2-bias.t7 b/models/modelL6_C-2-bias.t7
diff --git a/models/modelL6_C-2-weight.t7 b/models/modelL6_C-2-weight.t7
diff --git a/models/modelL6_C-3-bias.t7 b/models/modelL6_C-3-bias.t7
diff --git a/models/modelL6_C-3-weight.t7 b/models/modelL6_C-3-weight.t7
diff --git a/models/modelL6_C-4-bias.t7 b/models/modelL6_C-4-bias.t7
diff --git a/models/modelL6_C-4-weight.t7 b/models/modelL6_C-4-weight.t7
diff --git a/models/modelL6_C-5-bias.t7 b/models/modelL6_C-5-bias.t7
diff --git a/models/modelL6_C-5-weight.t7 b/models/modelL6_C-5-weight.t7
diff --git a/models/modelL6_F-1-bias.t7 b/models/modelL6_F-1-bias.t7
diff --git a/models/modelL6_F-1-weight.t7 b/models/modelL6_F-1-weight.t7
diff --git a/models/modelL6_F-2-bias.t7 b/models/modelL6_F-2-bias.t7
diff --git a/models/modelL6_F-2-weight.t7 b/models/modelL6_F-2-weight.t7
diff --git a/models/modelL6_F-3-bias.t7 b/models/modelL6_F-3-bias.t7
diff --git a/models/modelL6_F-3-weight.t7 b/models/modelL6_F-3-weight.t7
diff --git a/models/modelL6_F-4-bias.t7 b/models/modelL6_F-4-bias.t7
diff --git a/models/modelL6_F-4-weight.t7 b/models/modelL6_F-4-weight.t7
diff --git a/models/modelL6_F-5-bias.t7 b/models/modelL6_F-5-bias.t7
diff --git a/models/modelL6_F-5-weight.t7 b/models/modelL6_F-5-weight.t7
diff --git a/models/modelL6_K-1-bias.t7 b/models/modelL6_K-1-bias.t7
diff --git a/models/modelL6_K-1-weight.t7 b/models/modelL6_K-1-weight.t7
diff --git a/models/modelL6_K-2-bias.t7 b/models/modelL6_K-2-bias.t7
diff --git a/models/modelL6_K-2-weight.t7 b/models/modelL6_K-2-weight.t7
diff --git a/models/modelL6_K-3-bias.t7 b/models/modelL6_K-3-bias.t7
diff --git a/models/modelL6_K-3-weight.t7 b/models/modelL6_K-3-weight.t7
diff --git a/models/modelL6_K-4-bias.t7 b/models/modelL6_K-4-bias.t7
diff --git a/models/modelL6_K-4-weight.t7 b/models/modelL6_K-4-weight.t7
diff --git a/models/modelL6_K-5-bias.t7 b/models/modelL6_K-5-bias.t7
diff --git a/models/modelL6_K-5-weight.t7 b/models/modelL6_K-5-weight.t7
diff --git a/run.py b/run.py
@@ -35,23 +35,6 @@
 	if strOption == '--out' and strArgument != '': arguments_strOut = strArgument # path to where the output should be stored
 # end
 
-if arguments_strModel == 'chairs-clean':
-	arguments_strModel = '4'
-
-elif arguments_strModel == 'chairs-final':
-	arguments_strModel = '3'
-
-elif arguments_strModel == 'sintel-clean':
-	arguments_strModel = 'C'
-
-elif arguments_strModel == 'sintel-final':
-	arguments_strModel = 'F'
-
-elif arguments_strModel == 'kitti-final':
-	arguments_strModel = 'K'
-
-# end
-
 ##########################################################
 
 class Network(torch.nn.Module):
@@ -87,17 +70,6 @@ def __init__(self, intLevel):
 					torch.nn.ReLU(inplace=False),
 					torch.nn.Conv2d(in_channels=16, out_channels=2, kernel_size=7, stride=1, padding=3)
 				)
-
-				if intLevel == 5:
-					if arguments_strModel == '3' or arguments_strModel == '4':
-						intLevel = 4 # the models trained on the flying chairs dataset do not come with weights for the sixth layer
-					# end
-				# end
-
-				for intConv in range(5):
-					self.moduleBasic[intConv * 2].weight.data.copy_(torch.utils.serialization.load_lua('./models/modelL' + str(intLevel + 1) + '_' + arguments_strModel  + '-' + str(intConv + 1) + '-weight.t7'))
-					self.moduleBasic[intConv * 2].bias.data.copy_(torch.utils.serialization.load_lua('./models/modelL' + str(intLevel + 1) + '_' + arguments_strModel  + '-' + str(intConv + 1) + '-bias.t7'))
-				# end
 			# end
 
 			def forward(self, tensorInput):
@@ -129,6 +101,8 @@ def forward(self, tensorInput, tensorFlow):
 		self.moduleBasic = torch.nn.ModuleList([ Basic(intLevel) for intLevel in range(6) ])
 
 		self.moduleBackward = Backward()
+
+		self.load_state_dict(torch.load('./network-' + arguments_strModel + '.pytorch'))
 	# end
 
 	def forward(self, tensorFirst, tensorSecond):