Skip to content

Commit

Permalink
README.md for train (experimental)
Browse files Browse the repository at this point in the history
  • Loading branch information
longcw committed Feb 12, 2017
1 parent b5772e6 commit e00d19a
Showing 1 changed file with 21 additions and 1 deletion.
22 changes: 21 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ by Shaoqing Ren, Kaiming He, Ross Girshick, Jian Sun.
- [x] roi pooling layer implemented by python and pytorch
- [x] roi pooling layer with C extensions on CPU (only forward)
- [x] roi pooling layer on GPU (forward and backward)
- [ ] backward pass for training
- [x] backward pass for training (experimental)

### Installation and demo
1. Clone the Faster R-CNN repository
Expand All @@ -32,3 +32,23 @@ by Shaoqing Ren, Kaiming He, Ross Girshick, Jian Sun.
and set the model path in `demo.py`
3. Run demo `python demo.py`

### Training on Pascal VOC 2007
**NOTE: The training method in this project is still experimental.**

Follow [this project (TFFRCNN)](https://github.com/CharlesShang/TFFRCNN)
to download and prepare the training, validation, test data
and the VGG16 model pre-trained on ImageNet.

Since the program loading the data in `faster_rcnn_pytorch/data` by default,
you can set the data path as following.
```bash
cd faster_rcnn_pytorch
mkdir data
cd data
ln -s $VOCdevkit VOCdevkit2007
```

Then you can set some hyper-parameters in `train.py` and training parameters in the `.yml` file.

You may need to train RPN and the classifier separately as described in the Faster RCNN paper
and tune the loss function defined in `faster_rcnn/faster_rcnn.py` by yourself.

0 comments on commit e00d19a

Please sign in to comment.