Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
Zhaowei Cai authored Aug 23, 2016
1 parent 0eacba9 commit 610f67e
Showing 1 changed file with 9 additions and 9 deletions.
18 changes: 9 additions & 9 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ If you use our code/model/data, please cite our paper:

1. cuDNN is required to have the same running speed described in our paper. For now, only cuDNN 7.0 is supported.

2. If you want to use our MATLAB scripts to run the detection demo, MATLAB wrapper is required. Please build matcaffe before running the detection demo.
2. If you want to use our MATLAB scripts to run the detection demo, caffe MATLAB wrapper is required. Please build matcaffe before running the detection demo.

3. This code has been tested on Ubuntu 14.04 with an NVIDIA Titan GPU.

Expand All @@ -50,24 +50,24 @@ If you use our code/model/data, please cite our paper:

### Training MS-CNN (KITTI car)

1. Set up KITTI dataset by yourself
1. Set up KITTI dataset by yourself.

2. Get the training data for KITTI
```Shell
cd $MSCNN_ROOT/data/
sh get_kitti_data.sh
```

This will download train/val split image lists for the experiments, and window files for training/finetuning MS-CNN models. You can also use the provided MATLAB scripts to generate your own window files. If you use the provided window files, replace `/your/KITTI/path/` in the files to your KITTI path.
This will download train/val split image lists for the experiments, and window files for training/finetuning MS-CNN models. You can also use the provided MATLAB scripts `mscnn_kitti_car_window_file.m` under `$MSCNN_ROOT/data/kitti/` to generate your own window files. If you use the provided window files, replace `/your/KITTI/path/` in the files to your KITTI path.

3. Download VGG16 from [Caffe Model Zoo](https://github.com/BVLC/caffe/wiki/Model-Zoo), and put it into `$MSCNN_ROOT/models/VGG/`
3. Download VGG16 from [Caffe Model Zoo](https://github.com/BVLC/caffe/wiki/Model-Zoo), and put it into `$MSCNN_ROOT/models/VGG/`.

4. Now you can start to train MS-CNN models. Multiple shell scripts are provided to train different models described in our paper. We take `mscnn-7s-576-2x-train` for example.
4. Now you can start to train MS-CNN models. Multiple shell scripts are provided to train different models described in our paper. We take `mscnn-7s-576-2x` for example.
```Shell
cd $MSCNN_ROOT/examples/kitti_car/mscnn-7s-576-2x-train/
cd $MSCNN_ROOT/examples/kitti_car/mscnn-7s-576-2x/
sh train_mscnn.sh
```
As described in the paper, the training process is split into two steps. Usually the first step can be shared by different models if you only have modifications on detection sub-network. For example, the first training step can be shared by `mscnn-7s-576-2x-train` and `mscnn-7s-576-train`. Meanwhile, log files will be generated along the training procedures.
As described in the paper, the training process is split into two steps. Usually the first step can be shared by different models if you only have modifications on detection sub-network. For example, the first training step can be shared by `mscnn-7s-576-2x` and `mscnn-7s-576`. Meanwhile, log files will be generated along the training procedures.


### Pretrained model (KITTI car)
Expand All @@ -81,13 +81,13 @@ This will download the pretrained model for KITTI car into `$MSCNN_ROOT/examples

### Testing Demo (KITTI car)

Once the pretrained models or models trained by yourself are available, you can use the MATLAB script `run_mscnn_detection.m` under `$MSCNN_ROOT/examples/kitti_car/` to obtain the detection and proposal results. Set the right dataset path and choose the model that you want to test in the demo scripts. The default setting is to test the pretrained model. The final results will be saved as .txt files.
Once the pretrained models or models trained by yourself are available, you can use the MATLAB script `run_mscnn_detection.m` under `$MSCNN_ROOT/examples/kitti_car/` to obtain the detection and proposal results. Set the right dataset path and choose the model that you want to test in the demo script. The default setting is to test the pretrained model. The final results will be saved as .txt files.

### Disclaimer

1. This implementation is developed on an old version of Caffe. Some new layers of current Caffe may not be supported in this implementation. But some layers (e.g. "batch_norm", "scale", "bias") are merged into this repository, such that ResNet is supported.

2. The CPU version is not fully tested. GPU version is strongly recommended.
2. The CPU version is not fully tested. The GPU version is strongly recommended.

3. Since some changes have been made after ECCV submission, you may not have exactly the same results in the paper by training your own models. But you should have equivelant performance.

Expand Down

0 comments on commit 610f67e

Please sign in to comment.