Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
haofengac authored Mar 12, 2019
1 parent 2773e4a commit 534ec22
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Faster R-CNN for KITTI and BDD100k
# Faster R-CNN for KITTI and BDD100K

This project is based on facebookresearch's [maskrcnn-benchmark](https://github.com/facebookresearch/maskrcnn-benchmark).

Expand Down Expand Up @@ -30,7 +30,7 @@ pip install tensorboardX
### KITTI
This repo assumes that you followed NVIDIA DIGIT's instruction to prepare the KITTI dataset.

### BDD100k
### BDD100K
Prepare the dataset with the below structure:
```
bdd100k/
Expand All @@ -56,12 +56,12 @@ ln -s {parent directory of kitti and bdd100k} {root directory of this repo}/data
python tools/train_net.py --config-file "configs/e2e_faster_rcnn_R_101_FPN_1x.yaml" --use_tensorboard MODEL.ROI_BOX_HEAD.NUM_CLASSES 7 DATASETS.TRAIN '("kitti_train",)' OUTPUT_DIR out_kitti
```

### BDD100k
### BDD100K
```
python tools/train_net.py --config-file "configs/e2e_faster_rcnn_R_101_FPN_1x.yaml" --use_tensorboard MODEL.ROI_BOX_HEAD.NUM_CLASSES 11 DATASETS.TRAIN '("bdd100k_train",)' OUTPUT_DIR out_bdd
```

For the BDD100k dataset, you might need to clip the gradient for training stability. For other hyperparameters, refer to [maskrcnn-benchmark](https://github.com/facebookresearch/maskrcnn-benchmark).
For the BDD100K dataset, you might need to clip the gradient for training stability. For other hyperparameters, refer to [maskrcnn-benchmark](https://github.com/facebookresearch/maskrcnn-benchmark).


## Inference
Expand All @@ -73,7 +73,7 @@ Below commands loads saved weights and save the detections in the [BDD format](h
python tools/test_net.py --weights out_kitti/*.pth --config-file "configs/e2e_faster_rcnn_R_101_FPN_1x.yaml" DATASETS.TEST '("kitti_val",)' OUTPUT_DIR out_kitti
```

### BDD100k
### BDD100K
```
python tools/test_net.py --weights out_bdd/*.pth --config-file "configs/e2e_faster_rcnn_R_101_FPN_1x.yaml" DATASETS.TEST '("bdd100k_val",)' OUTPUT_DIR out_bdd
```
Expand Down

0 comments on commit 534ec22

Please sign in to comment.