Skip to content

Commit 7b5f739

Browse files
Update README.md
1 parent a98105f commit 7b5f739

File tree

1 file changed

+22
-3
lines changed

1 file changed

+22
-3
lines changed

README.md

+22-3
Original file line numberDiff line numberDiff line change
@@ -22,12 +22,31 @@ cd ..
2222
### Data preparation
2323
We convert [ScanNet](http://www.scan-net.org/) data to *.tfrecords* files for training and testing. The *.tfrecords* file can be downloaded from [here](https://mega.nz/#!IvAixABb!PD3wJtXX_6W3qtfKZQtl_P07mYPLwWst3cwbvuTXlSY).
2424

25-
### Train the network
25+
### Training
2626
To train the network from the pretrained DeepLab network, please first download the DeepLab model [here](https://github.com/DrSleep/tensorflow-deeplab-resnet) (under the Caffe to TensorFlow conversion), and then run the following command.
27-
2827
```bash
2928
python train_planenet.py --restore=0 --modelPathDeepLab="path to the deep lab model" --rootFolder="folder which contains tfrecords files"
3029
```
3130

3231
### Evaluation
33-
Please first download our trained network from [here](https://mega.nz/#!sjpT2DiQ!Uo-6hxyldmtnPoKk3TTdUHKZADRGy6nIPlmAeVzJs_8) and put the uncompressed folder under ./checkpoint
32+
Please first download our trained network from [here](https://mega.nz/#!sjpT2DiQ!Uo-6hxyldmtnPoKk3TTdUHKZADRGy6nIPlmAeVzJs_8) and put the uncompressed folder under ./checkpoint folder.
33+
34+
To evaluate the performance against existing methods, please run:
35+
```bash
36+
python evaluate.py --rootFolder="folder which contains tfrecords files"
37+
```
38+
39+
### Applications
40+
Please first download our trained network (see [Evaluation](### Evaluation) section for details). Script *predict.py* and *predict_custom.py* are for ScanNet testing images and custom images respectively. To predict and visualize ScanNet testing images, please run:
41+
42+
```bash
43+
python predict.py --rootFolder="folder which contains tfrecords files" [--startIndex=0] [--numImages=30]
44+
```
45+
46+
This will generate visualization images, a webpage containing all the visualization, as well as cache files under folder "predict". Similarly, the following command can be used to predict and visualize custom images:
47+
48+
```bash
49+
python predict_custom.py --rootFolder="folder which contains custom images"
50+
```
51+
52+
Same commands can be used for various applications by providing arguments, *--imageIndex*, *--suffix*, and *--textureFilename*

0 commit comments

Comments
 (0)