summaryrefslogtreecommitdiff
path: root/Codes
diff options
context:
space:
mode:
authorStevenLiuWen <liuwen@shanghaitech.edu.cn>2018-03-13 05:38:21 -0400
committerStevenLiuWen <liuwen@shanghaitech.edu.cn>2018-03-13 05:38:21 -0400
commit596fe06acf4c5382bd48aa5ae3809449cdb0532a (patch)
treebd5048ce403c318befdc2acd2578dda743f8f886 /Codes
parent4ad77cabf91f25d9bac7cc94bb8c324ee0cd1630 (diff)
update
Diffstat (limited to 'Codes')
-rw-r--r--Codes/constant.py2
-rw-r--r--Codes/flownet2/.gitignore9
2 files changed, 10 insertions, 1 deletions
diff --git a/Codes/constant.py b/Codes/constant.py
index eafeab9..f6f78df 100644
--- a/Codes/constant.py
+++ b/Codes/constant.py
@@ -92,7 +92,7 @@ const.EVALUATE = args.evaluate
# network constants
const.HEIGHT = 256
const.WIDTH = 256
-const.FLOWNET_CHECKPOINT = 'flownet2/checkpoints/FlowNetSD/flownet-SD.ckpt-0'
+const.FLOWNET_CHECKPOINT = 'models/pretrains/flownet-SD.ckpt-0'
const.FLOW_HEIGHT = 384
const.FLOW_WIDTH = 512
diff --git a/Codes/flownet2/.gitignore b/Codes/flownet2/.gitignore
new file mode 100644
index 0000000..31abf4e
--- /dev/null
+++ b/Codes/flownet2/.gitignore
@@ -0,0 +1,9 @@
+__pycache__/
+*.py[cod]
+*$py.class
+*.o
+*.so
+*.so.dSYM
+checkpoints/
+!checkpoints/download.sh
+!checkpoints/README.md