Skip to content

Commit

Permalink
Add an example for process.py
Browse files Browse the repository at this point in the history
  • Loading branch information
zhou13 committed Jan 27, 2020
1 parent 5d2bd1b commit fd4e2fc
Showing 1 changed file with 9 additions and 2 deletions.
11 changes: 9 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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.

Expand All @@ -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
Expand Down

0 comments on commit fd4e2fc

Please sign in to comment.