diff --git a/README.md b/README.md index 82d86ee..08052df 100644 --- a/README.md +++ b/README.md @@ -29,11 +29,11 @@ Following models are implemented, each one also has a readme inside. |[lenet](./lenet) | the simplest, as a "hello world" of this project | |[alexnet](./alexnet)| easy to implement, all layers are supported in tensorrt | |[googlenet](./googlenet)| GoogLeNet (Inception v1) | -|[inception](./inception)| Inception v3 | +|[inception](./inceptionv3)| Inception v3 | |[mnasnet](./mnasnet)| MNASNet with depth multiplier of 0.5 from the paper | -|[mobilenet](./mobilenet)| MobileNet V2 | +|[mobilenet](./mobilenetv2)| MobileNet V2 | |[resnet](./resnet)| resnet-18 and resnet-50 are implemented | -|[shufflenet](./shufflenet)| ShuffleNetV2 with 0.5x output channels | +|[shufflenet](./shufflenetv2)| ShuffleNetV2 with 0.5x output channels | |[squeezenet](./squeezenet)| SqueezeNet 1.1 model | |[vgg](./vgg)| VGG 11-layer model | |[yolov3](./yolov3)| darknet-53, weights from yolov3 authors | diff --git a/inception/CMakeLists.txt b/inceptionv3/CMakeLists.txt similarity index 100% rename from inception/CMakeLists.txt rename to inceptionv3/CMakeLists.txt diff --git a/inception/README.md b/inceptionv3/README.md similarity index 100% rename from inception/README.md rename to inceptionv3/README.md diff --git a/inception/common.h b/inceptionv3/common.h similarity index 100% rename from inception/common.h rename to inceptionv3/common.h diff --git a/inception/inception_v3.cpp b/inceptionv3/inception_v3.cpp similarity index 100% rename from inception/inception_v3.cpp rename to inceptionv3/inception_v3.cpp diff --git a/mobilenet/CMakeLists.txt b/mobilenetv2/CMakeLists.txt similarity index 100% rename from mobilenet/CMakeLists.txt rename to mobilenetv2/CMakeLists.txt diff --git a/mobilenet/README.md b/mobilenetv2/README.md similarity index 100% rename from mobilenet/README.md rename to mobilenetv2/README.md diff --git a/mobilenet/common.h b/mobilenetv2/common.h similarity index 100% rename from mobilenet/common.h rename to mobilenetv2/common.h diff --git a/mobilenet/mobilenet_v2.cpp b/mobilenetv2/mobilenet_v2.cpp similarity index 100% rename from mobilenet/mobilenet_v2.cpp rename to mobilenetv2/mobilenet_v2.cpp diff --git a/mobilenetv3/README.md b/mobilenetv3/README.md index 1210419..3ca490a 100644 --- a/mobilenetv3/README.md +++ b/mobilenetv3/README.md @@ -1,23 +1,23 @@ -# mobilenet v2 +# mobilenet v3 -MobileNetV2 architecture from - "MobileNetV2: Inverted Residuals and Linear Bottlenecks" . +MobileNetV3 architecture from + "Searching for MobileNetV3" . -For the Pytorch implementation, you can refer to [pytorchx/mobilenet](https://github.com/wang-xinyu/pytorchx/tree/master/mobilenet) +For the Pytorch implementation, you can refer to [mobilenetv3.pytorch](https://github.com/d-li14/mobilenetv3.pytorch) Following tricks are used in this mobilenet, -- Relu6 is used in mobilenet v2. We use `Relu6(x) = Relu(x) - Relu(x-6)` in tensorrt. +- Hsigmoid is used in mobilenet v3. We create a plugin in tensorrt. - Batchnorm layer, implemented by scale layer. ``` -// 1. generate mobilenet.wts from [pytorchx/mobilenet](https://github.com/wang-xinyu/pytorchx/tree/master/mobilenet) +// 1. generate mbv3_small.wts/mbv3_large.wts from pytorch implementation -// 2. put mobilenet.wts into tensorrtx/mobilenet +// 2. put mbv3_small.wts/mbv3_large.wts into tensorrtx/mobilenetv3 // 3. build and run -cd tensorrtx/mobilenet +cd tensorrtx/mobilenetv3 mkdir build @@ -27,11 +27,11 @@ cmake .. make -sudo ./mobilenet -s // serialize model to plan file i.e. 'mobilenet.engine' +sudo ./mobilenetv3 -s small(or large) // serialize model to plan file i.e. 'mobilenetv3_small.engine' -sudo ./mobilenet -d // deserialize plan file and run inference +sudo ./mobilenetv3 -d small(or large) // deserialize plan file and run inference -// 4. see if the output is same as pytorchx/mobilenet +// 4. see if the output is same as pytorch implementation ``` diff --git a/shufflenet/CMakeLists.txt b/shufflenetv2/CMakeLists.txt similarity index 100% rename from shufflenet/CMakeLists.txt rename to shufflenetv2/CMakeLists.txt diff --git a/shufflenet/README.md b/shufflenetv2/README.md similarity index 100% rename from shufflenet/README.md rename to shufflenetv2/README.md diff --git a/shufflenet/chunk.cpp b/shufflenetv2/chunk.cpp similarity index 100% rename from shufflenet/chunk.cpp rename to shufflenetv2/chunk.cpp diff --git a/shufflenet/chunk.h b/shufflenetv2/chunk.h similarity index 100% rename from shufflenet/chunk.h rename to shufflenetv2/chunk.h diff --git a/shufflenet/common.h b/shufflenetv2/common.h similarity index 100% rename from shufflenet/common.h rename to shufflenetv2/common.h diff --git a/shufflenet/shufflenet_v2.cpp b/shufflenetv2/shufflenet_v2.cpp similarity index 100% rename from shufflenet/shufflenet_v2.cpp rename to shufflenetv2/shufflenet_v2.cpp