Skip to content

Commit ba990b9

Browse files
authored
Update README.md
1 parent f813098 commit ba990b9

File tree

1 file changed

+10
-14
lines changed

1 file changed

+10
-14
lines changed

senet/README.md

Lines changed: 10 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1,23 +1,20 @@
1-
# mobilenet v2
1+
# SENet
22

3-
MobileNetV2 architecture from
4-
"MobileNetV2: Inverted Residuals and Linear Bottlenecks" <https://arxiv.org/abs/1801.04381>.
3+
An implementation of SENet, proposed in Squeeze-and-Excitation Networks by Jie Hu, Li Shen, Samuel Albanie, Gang Sun, Enhua Wu
54

6-
For the Pytorch implementation, you can refer to [pytorchx/mobilenet](https://github.com/wang-xinyu/pytorchx/tree/master/mobilenet)
5+
[https://arxiv.org/abs/1709.01507](https://arxiv.org/abs/1709.01507)
76

8-
Following tricks are used in this mobilenet,
7+
For the Pytorch implementation, you can refer to [wang-xinyu/senet.pytorch](https://github.com/wang-xinyu/senet.pytorch).
98

10-
- Relu6 is used in mobilenet v2. We use `Relu6(x) = Relu(x) - Relu(x-6)` in tensorrt.
11-
- Batchnorm layer, implemented by scale layer.
129

1310
```
14-
// 1. generate mobilenet.wts from [pytorchx/mobilenet](https://github.com/wang-xinyu/pytorchx/tree/master/mobilenet)
11+
// 1. generate se_resnet50.wts from [wang-xinyu/senet.pytorch](https://github.com/wang-xinyu/senet.pytorch)
1512
16-
// 2. put mobilenet.wts into tensorrtx/mobilenet
13+
// 2. put se_resnet50.wts into tensorrtx/senet
1714
1815
// 3. build and run
1916
20-
cd tensorrtx/mobilenet
17+
cd tensorrtx/senet
2118
2219
mkdir build
2320
@@ -27,11 +24,10 @@ cmake ..
2724
2825
make
2926
30-
sudo ./mobilenet -s // serialize model to plan file i.e. 'mobilenet.engine'
27+
sudo ./se_resnet -s // serialize model to plan file i.e. 'se_resnet50.engine'
3128
32-
sudo ./mobilenet -d // deserialize plan file and run inference
29+
sudo ./se_resnet -d // deserialize plan file and run inference
3330
34-
// 4. see if the output is same as pytorchx/mobilenet
31+
// 4. see if the output is same as [wang-xinyu/senet.pytorch]
3532
```
3633

37-

0 commit comments

Comments
 (0)