Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
coldmanck committed Nov 27, 2015
1 parent c676083 commit ab76e4f
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -212,11 +212,12 @@ Run `$ export LD_LIBRARY_PATH=/usr/local/cuda/lib64` in every session (every con

**Error message** `Check failed: ShapeEquals(proto) shape mismatch(reshape not set)`

First, the answer in stackoverflow saying that it's may caused by forgetting to change the number of classes in `test.prototxt` (same as `train.protxt`) to fit your dataset. If you've changed it but still encounterd the error, follow my renaming approach (rename layers `bbox_pred` and `cls_score`) descripted in this article to re-train your model.
First, one of the [answer](http://stackoverflow.com/a/31251378/4447620) in stackoverflow saying that it's may caused by forgetting to change the number of classes in `test.prototxt` (same as `train.protxt`) to fit your dataset. If you've changed it but still encounterd the error, follow my renaming approach (rename layers `bbox_pred` and `cls_score`) descripted in this article to re-train your model.

## Reference
1. [Train Fast-RCNN on Another Dataset](https://github.com/zeyuanxy/fast-rcnn/tree/master/help/train)
2. [How to train fast rcnn on imagenet](http://sunshineatnoon.github.io/Train-fast-rcnn-model-on-imagenet-without-matlab/)
3. [Selective Search Configuration](https://github.com/rbgirshick/fast-rcnn/issues/26)
4. [Fast rcnn 訓練自己的數據庫問題小結](http://blog.csdn.net/hao529good/article/details/46544163)
5. [How to forward X over SSH from Ubuntu machine?](http://unix.stackexchange.com/questions/12755/how-to-forward-x-over-ssh-from-ubuntu-machine)
6. [[Caffe]: Check failed: ShapeEquals(proto) shape mismatch (reshape not set)](http://stackoverflow.com/a/31251378/4447620)

0 comments on commit ab76e4f

Please sign in to comment.