Skip to content

Commit

Permalink
Update readme.md
Browse files Browse the repository at this point in the history
  • Loading branch information
iamhankai authored Apr 7, 2021
1 parent 2063d0f commit 5a06c87
Showing 1 changed file with 2 additions and 5 deletions.
7 changes: 2 additions & 5 deletions ghostnet_pytorch/readme.md
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ By Kai Han, Yunhe Wang, Qi Tian, Jianyuan Guo, Chunjing Xu, Chang Xu.

- **Performance**

GhostNet beats other SOTA lightweight CNNs such as **MobileNetV3** and **FBNet**.
GhostNet beats other SOTA lightweight CNNs such as MobileNetV3 and FBNet.

<div align="center">
<img src="../fig/flops_latency.png" width="720">
Expand All @@ -28,7 +28,7 @@ This folder provides the PyTorch code and pretrained model of GhostNet on ImageN
The code was verified on Python3.6, PyTorch-1.0+.

### Usage
Run `python validate.py --eval --data=/path/to/imagenet/dir/` to evaluate on `val` set.
Run `python validate.py --data=/path/to/imagenet/dir/` to evaluate on `val` set.

You'll get the accuracy: top-1 acc=`0.7398` and top-5 acc=`0.9146` with only `142M` Flops (or say MAdds).

Expand All @@ -43,10 +43,7 @@ dir/
ILSVRC2012_val_00000293.JPEG
...
...
caffe_ilsvrc12/
...
```
caffe_ilsvrc12 data can be downloaded from http://dl.caffe.berkeleyvision.org/caffe_ilsvrc12.tar.gz

## Citation
```
Expand Down

0 comments on commit 5a06c87

Please sign in to comment.