add G-GhostNet

huawei-noah · Nov 13, 2022 · 9e0b8ab · 9e0b8ab
1 parent 84bf6d3
commit 9e0b8ab
Show file tree

Hide file tree

Showing 5 changed files with 44 additions and 7 deletions.
diff --git a/fig/g-ghost.png b/fig/g-ghost.png
diff --git a/ghostnet_pytorch/g_ghost_regnet.py → g_ghostnet_pytorch/g_ghost_regnet.py b/ghostnet_pytorch/g_ghost_regnet.py → g_ghostnet_pytorch/g_ghost_regnet.py
diff --git a/g_ghostnet_pytorch/readme.md b/g_ghostnet_pytorch/readme.md
@@ -0,0 +1,43 @@
+# G-GhostNet
+
+GhostNets on Heterogeneous Devices via Cheap Operations. IJCV 2022. [[arXiv]](https://arxiv.org/abs/2201.03297)
+
+By Kai Han, Yunhe Wang, Chang Xu, Jianyuan Guo, Chunjing Xu, Enhua Wu and Qi Tian.
+
+- **Approach**
+
+<div align="center">
+   <img src="../fig/g-ghost.png" width="720">
+</div>
+
+## Requirements
+The code was verified on Python3.6, PyTorch-1.3+.
+
+## Data Preparation
+ImageNet data dir should have the following structure, and `val` and `caffe_ilsvrc12` subdirs are essential:
+```
+dir/
+  train/
+    ...
+  val/
+    n01440764/
+      ILSVRC2012_val_00000293.JPEG
+      ...
+    ...
+```
+
+## G-GhostNet Code
+
+`g_ghost_regnet.py` implemented `G-Ghost RegNet` in the extended [IJCV paper](https://arxiv.org/abs/2201.03297).
+
+The checkpoint is available at [BaiduDisk](https://pan.baidu.com/s/1bgdM9xWVCFGMyYMKPUn4Zg) and Password: `on8i`.
+
+## Citation
+```
+@article{ghostnet2022,
+  title={GhostNets on Heterogeneous Devices via Cheap Operations},
+  author={Kai Han, Yunhe Wang, Chang Xu, Jianyuan Guo, Chunjing Xu, Enhua Wu and Qi Tian},
+  journal={IJCV},
+  year={2022}
+}
+```
diff --git a/ghostnet_pytorch/readme.md b/ghostnet_pytorch/readme.md
@@ -45,12 +45,6 @@ dir/
     ...
 ```
 
-# G-GhostNet
-
-`g_ghost_regnet.py` implemented `G-Ghost RegNet` in the extended [IJCV paper](https://arxiv.org/abs/2201.03297).
-
-The checkpoint is available at [BaiduDisk](https://pan.baidu.com/s/1bgdM9xWVCFGMyYMKPUn4Zg) and Password: `on8i`.
-
 ## Citation
 ```
 @inproceedings{ghostnet2020,

diff --git a/vig_pytorch/readme.md b/vig_pytorch/readme.md
@@ -14,7 +14,7 @@ apex
 - ViG
 
 |Model|Params (M)|FLOPs (B)|Top-1|Github Release|
-|-|-|-|-|-|-|
+|-|-|-|-|-|
 |ViG-Ti|7.1|1.3|73.9|[Github Release](https://github.com/huawei-noah/Efficient-AI-Backbones/releases/tag/vig)|
 |ViG-S|22.7|4.5|80.4|[Github Release](https://github.com/huawei-noah/Efficient-AI-Backbones/releases/tag/vig)|
 |ViG-B|86.8|17.7|82.3|[Github Release](https://github.com/huawei-noah/Efficient-AI-Backbones/releases/tag/vig)|