Browse Source

Add an example for process.py

Yichao Zhou 6 years ago
parent
commit
fd4e2fcc6b
1 changed files with 9 additions and 2 deletions
  1. 9 2
      README.md

+ 9 - 2
README.md

@@ -114,7 +114,7 @@ cd ..
 ```
 
 If `gdrive-download.sh` does not work for you, you can download the pre-processed dataset
-`wireframe.tar.xz`  manually from [Google
+`wireframe.tar.xz` manually from [Google
 Drive](https://drive.google.com/drive/u/1/folders/1rXLAh5VIj8jwf8vLfuZncStihRO2chFr) and proceed
 accordingly.
 
@@ -135,7 +135,14 @@ dataset/wireframe.py data/wireframe_raw data/wireframe
 ### Training
 To train the neural network on GPU 0 (specified by `-d 0`) with the default parameters, execute
 ```bash
-python ./train.py -d 0 --identifier baseline  config/wireframe.yaml
+python ./train.py -d 0 --identifier baseline config/wireframe.yaml
+```
+
+## Testing Pretrained Models
+To generate wireframes on the validation dataset with the pretrained model, execute
+
+```bash
+./process.py config/wireframe.yaml <path-to-checkpoint.pth.tar> data/wireframe logs/pretrained-model/npz/000312000
 ```
 
 ### Post Processing