From 26c99591b65678cf1f4ab1691347d388d39c5b9f Mon Sep 17 00:00:00 2001 From: linghu8812 Date: Fri, 13 Nov 2020 22:50:23 +0800 Subject: [PATCH] update README.md --- README.md | 5 +++-- ghostnet/README.md | 12 +++++++----- 2 files changed, 10 insertions(+), 7 deletions(-) diff --git a/README.md b/README.md index 5b0b5ebf..919170c6 100644 --- a/README.md +++ b/README.md @@ -9,12 +9,13 @@ models|framework|instruction ---|---|--- [lenet](lenet)|PyTorch|An example from model training to TensorRT model deploy [alexnet](alexnet)|MXNet Gluon|MXNet Gluon example -[seresnext](seresnext)|PyTorch|PyTorch example [arcface](arcface)|MXNet Symbol|MXNet Symbol and face recognition example +[CenterFace](CenterFace)|ONNX|rewrite ONNX model and face detection example [face_alignment](face_alignment)|MXNet Symbol|MXNet Symbol and face key points detection example [gender-age](gender-age)|MXNet Symbol|MXNet Symbol and face gender and age recognize example +[ghostnet](ghostnet)|PyTorch|PyTorch example [RetinaFace](RetinaFace)|MXNet Symbol|MXNet Symbol and face detection example -[CenterFace](CenterFace)|ONNX|rewrite ONNX model and face detection example +[seresnext](seresnext)|PyTorch|PyTorch example [Yolov4](Yolov4)|darknet|darknet and object detection example [yolov5](yolov5)|PyTorch|PyTorch and object detection example \ No newline at end of file diff --git a/ghostnet/README.md b/ghostnet/README.md index d2e44415..1e675b79 100644 --- a/ghostnet/README.md +++ b/ghostnet/README.md @@ -1,21 +1,23 @@ -# AlexNet MXNet Gluon=>ONNX=>TensorRT +# ghostnet PyTorch=>ONNX=>TensorRT ## 1.Reference -- **AlexNet:** [Imagenet classification with deep convolutional neural networks](https://papers.nips.cc/paper/4824-imagenet-classification-with-deep-convolutional-neural-networks.pdf) +- **arxiv:** [GhostNet: More Features from Cheap Operations](https://arxiv.org/abs/1911.11907) +- **github:** [https://github.com/huawei-noah/ghostnet](https://github.com/huawei-noah/ghostnet) +- get ghostnet weights from here: [ghostnet/pytorch](https://github.com/huawei-noah/ghostnet/blob/master/pytorch/models/state_dict_93.98.pth) ## 2.Export ONNX Model ``` python3 export_onnx.py ``` -## 3.Build alexnet_trt Project +## 3.Build ghostnet_trt Project ``` mkdir build && cd build cmake .. make -j ``` -## 4.run alexnet_trt +## 4.run ghostnet_trt ``` -./alexnet_trt ../config.yaml ../samples +./ghostnet_trt ../config.yaml ../samples ``` \ No newline at end of file