Skip to content

Commit 4d5f3bb

Browse files
Update README.md
1 parent 69c408f commit 4d5f3bb

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

README.md

+8-2
Original file line numberDiff line numberDiff line change
@@ -5,11 +5,17 @@ Python 2.7, TensorFlow (>= 1.0), numpy, opencv 3.
55

66
## Getting started
77
### Compilation
8-
Please run the following command to compile the library for computing the set matching loss. (See [here](https://github.com/fanhqme/PointSetGeneration) for details.)
8+
Please run the *make* command under the root folder to compile the library for computing the set matching loss. (See [here](https://github.com/fanhqme/PointSetGeneration) for details.)
99

10+
### Data preparation
11+
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).
1012

13+
### Train the network
1114
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.
1215

1316
```bash
14-
python train_planenet.py --restore=0
17+
python train_planenet.py --restore=0 --modelPathDeepLab="path to the deep lab model" --rootFolder="folder which contains tfrecords files"
1518
```
19+
20+
### Evaluation
21+
Please first download our trained network from [here](https://mega.nz/#!sjpT2DiQ!Uo-6hxyldmtnPoKk3TTdUHKZADRGy6nIPlmAeVzJs_8) and put the uncompressed folder under ./checkpoint

0 commit comments

Comments
 (0)