Skip to content

Commit

Permalink
Release matterportB checkpoints.
Browse files Browse the repository at this point in the history
  • Loading branch information
Kefan Chen authored and Kefan Chen committed Apr 20, 2022
1 parent 012747c commit 83ac360
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ python train.py \
--data_dir <path_to_training_set> \
--model 9D
```
2. Save the estimated rotations as pickle dictionary in the data directories respectively.
2. Run DirectionNet-R on the training and test sets, then save the estimated rotations as Python pickle dictionary in the data directories respectively.
3. train DirectionNet-T.
```
python train.py \
Expand All @@ -41,7 +41,7 @@ python train.py \
```

## Evaluation
#DirectionNet-R
DirectionNet-R
```
python eval.py \
--checkpoint_dir <path_to_load_checkpoints> \
Expand All @@ -52,7 +52,7 @@ python eval.py \
--model 9D
```

#DirectionNet-T
DirectionNet-T
```
python eval.py \
--checkpoint_dir <path_to_load_checkpoints> \
Expand Down

0 comments on commit 83ac360

Please sign in to comment.