Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
syb7573330 authored Feb 5, 2018
1 parent a273841 commit 9929957
Showing 1 changed file with 20 additions and 0 deletions.
20 changes: 20 additions & 0 deletions sem_seg/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -11,3 +11,23 @@ Download 3D indoor parsing dataset (<a href="http://buildingparser.stanford.edu/
python collect_indoor3d_data.py
```
to generate "dgcnn/data/stanford_indoor3d"

### Train

We use 6-fold training, such that 6 models are trained leaving 1 of 6 areas as the testing area for each model. We keep using 2 GPUs for distributed training. To train 6 models sequentially, run
```
sh +x train_job.sh
```

### Evaluation

1. To generate predicted results for all 6 areas, run
```
sh +x test_job.sh
```
The model parameters are saved every 10 epochs, the saved model used to generate predited results can be changed by setting "--model_path" in "test_job.sh". For example, if you want to use the model saved after 70 epochs, you can set "--model_path" to "log*n*/epoch_70.ckpt" for *n* = 1, 2, ..., 6. To visualize the results, you can add "--visu" flag in the end of each line in "test_job.sh".

2. To obtain overall quantitative evaluation results, run
```
python eval_iou_accuracy.py
```

0 comments on commit 9929957

Please sign in to comment.