diff --git a/README.md b/README.md index 6f7119c..5cb346f 100644 --- a/README.md +++ b/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 data/wireframe logs/pretrained-model/npz/000312000 ``` ### Post Processing