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
*Note*: **\*** The speed here is tested on the newest pytorch and cudnn version (0.2.0 and cudnnV6), which is obviously faster than the speed reported in the paper (using pytorch-0.1.12 and cudnnV5).
\*: slightly better than the original ones in the paper (20.5).
42
42
@@ -48,14 +48,15 @@
48
48
5.[Models](#models)
49
49
50
50
## Installation
51
-
- Install [PyTorch-0.2.0](http://pytorch.org/) by selecting your environment on the website and running the appropriate command.
52
-
- Clone this repository. This repository is mainly based on [ssd.pytorch](https://github.com/amdegroot/ssd.pytorch) and [Chainer-ssd](https://github.com/Hakuyume/chainer-ssd), a huge thank to them.
51
+
- Install [PyTorch-0.2.0+](http://pytorch.org/) by selecting your environment on the website and running the appropriate command.
52
+
- Clone this repository. This repository is mainly based on[RFBNet](https://github.com/ruinmessi/RFBNet),[ssd.pytorch](https://github.com/amdegroot/ssd.pytorch) and [Chainer-ssd](https://github.com/Hakuyume/chainer-ssd), a huge thank to them.
53
53
* Note: We currently only support Python 3+.
54
54
- Compile the nms and coco tools:
55
55
```Shell
56
56
./make.sh
57
57
```
58
-
*Note*: Check you GPU architecture support in utils/build.py, line 131. Default is:
58
+
Note*: Check you GPU architecture support in utils/build.py, line 131. Default is:
59
+
59
60
```
60
61
'nvcc': ['-arch=sm_52',
61
62
```
@@ -101,7 +102,7 @@ $COCO/images/val2014/
101
102
102
103
## Training
103
104
- First download the fc-reduced [VGG-16](https://arxiv.org/abs/1409.1556) PyTorch base network weights at: https://s3.amazonaws.com/amdegroot-models/vgg16_reducedfc.pth
104
-
or from our [BaiduYun Driver](https://pan.baidu.com/s/1jIP86jW)
105
+
or from our [BaiduYun Driver](https://pan.baidu.com/s/1jIP86jW)
105
106
- MobileNet pre-trained basenet is ported from [MobileNet-Caffe](https://github.com/shicai/MobileNet-Caffe), which achieves slightly better accuracy rates than the original one reported in the [paper](https://arxiv.org/abs/1704.04861), weight file is available at: https://drive.google.com/open?id=13aZSApybBDjzfGIdqN1INBlPsddxCK14 or [BaiduYun Driver](https://pan.baidu.com/s/1dFKZhdv).
106
107
107
108
- By default, we assume you have downloaded the file in the `RFBNet/weights` dir:
0 commit comments