You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: README.md
+17-7Lines changed: 17 additions & 7 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -1,3 +1,12 @@
1
+
# Important notice:
2
+
If you used the master branch before Sep. 26 2017 and its corresponding pretrained model, **PLEASE PAY ATTENTION**:
3
+
The old master branch in now under old_master, you can still run the code and download the pretrained model, but the pretrained model for that old master is not compatible to the current master!
4
+
5
+
The main differences between new and old master branch are in this two commits: [9d4c24e](https://github.com/ruotianluo/pytorch-faster-rcnn/commit/9d4c24e83c3e4ec33751e50d5e4d8b1dd793dfaa), [c899ce7](https://github.com/ruotianluo/pytorch-faster-rcnn/commit/c899ce70dae62e3db1a5805eda96df88e4b59ca6)
6
+
The change is related to this [issue](https://github.com/ruotianluo/pytorch-faster-rcnn/issues/6); master now matches all the details (of resnet101) in [tf-faster-rcnn](https://github.com/endernewton/tf-faster-rcnn) so that we can now convert pretrained tf model to pytorch model (only resnet101 is supported).
7
+
8
+
(vgg16 conversion is not among the top of my to-do list. May reach it some day.)
9
+
1
10
# pytorch-faster-rcnn
2
11
A pytorch implementation of faster RCNN detection framework based on Xinlei Chen's [tf-faster-rcnn](https://github.com/endernewton/tf-faster-rcnn). Xinlei Chen's repository is based on the python Caffe implementation of faster RCNN available [here](https://github.com/rbgirshick/py-faster-rcnn).
3
12
@@ -12,9 +21,9 @@ With VGG16 (``conv5_3``):
12
21
- Train on COCO 2014 [trainval35k](https://github.com/rbgirshick/py-faster-rcnn/tree/master/models) and test on [minival](https://github.com/rbgirshick/py-faster-rcnn/tree/master/models) (900k/1190k) **27.0** (**29.5** for tf-faster-rcnn). ((350k/490k) **24.6**(crop and resize) **21.8**(roi pooling)).
13
22
14
23
With Resnet101 (last ``conv4``):
15
-
- Train on VOC 2007 trainval and test on VOC 2007 test, **75.08**(converted) (**75.2** for tf-faster-rcnn).
16
-
- Train on VOC 2007+2012 trainval and test on VOC 2007 test (R-FCN schedule), **78.85**(converted) (**79.3** for tf-faster-rcnn).
17
-
- Train on COCO 2014 trainval35k and test on minival (800k/1190k), **34.0**(converted) (**34.1** for tf-faster-rcnn).
24
+
- Train on VOC 2007 trainval and test on VOC 2007 test, **74.84**(from scratch) **75.08**(converted) (**75.2** for tf-faster-rcnn).
25
+
- Train on VOC 2007+2012 trainval and test on VOC 2007 test (R-FCN schedule), **78.82**(from scratch) **78.85**(converted) (**79.3** for tf-faster-rcnn).
26
+
- Train on COCO 2014 trainval35k and test on minival (800k/1190k), **34.8**(from scratch) **34.0**(converted) (**34.1** for tf-faster-rcnn).
18
27
19
28
More Results:
20
29
- Train Mobilenet (1.0, 224) on COCO 2014 trainval35k and test on minival (900k/1190k), ~~**21.9**~~.
@@ -107,17 +116,18 @@ Please follow the instructions of py-faster-rcnn [here](https://github.com/rbgir
107
116
If you find it useful, the ``data/cache`` folder created on my side is also shared [here](http://ladoga.graphics.cs.cmu.edu/xinleic/tf-faster-rcnn/cache.tgz).
108
117
109
118
### Demo and Test with pre-trained models
110
-
~~1. Download pre-trained model (only google drive works)~~
119
+
1. Download pre-trained model (only google drive works)
111
120
<!-- ```Shell
112
121
# Resnet101 for voc pre-trained on 07+12 set
113
122
# ./data/scripts/fetch_faster_rcnn_models.sh
114
123
```
115
124
**Note**: if you cannot download the models through the link, or you want to try more models, you can check out the following solutions and optionally update the downloading script: -->
116
125
-~~Another server [here](http://gs11655.sp.cs.cmu.edu/xinleic/tf-faster-rcnn/).~~
- Google drive [here](https://drive.google.com/open?id=0B7fNdx_jAqhtNE10TDZDbFRuU0E).
118
127
119
-
1. Convert from tf-faster-rcnn model (Only support res101 currently).
120
-
You can download the pretrained model from [tf-faster-rcnn](https://github.com/endernewton/tf-faster-rcnn/#demo-and-test-with-pre-trained-models).
128
+
**(Optional)**
129
+
Instead of downloading my pretrained or converted model, you can also convert from tf-faster-rcnn model (Only support res101 currently).
130
+
You can download the tensorflow pretrained model from [tf-faster-rcnn](https://github.com/endernewton/tf-faster-rcnn/#demo-and-test-with-pre-trained-models).
0 commit comments