update README.md

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
 

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
 ```