SA-MobileNetV3

KO | EN

gitlite — search

#javascript #hacktoberfest #react #ai #go #rust #linux #docker #java #windows #awesome #awesome-list #pytorch #openai

(★ 52)

Shuffle Attention for MobileNetV3

SA-MobileNetV3 최신버젼 다운로드

최종 버전 다운로드 (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# SA-MobileNetV3

[![Python package](https://github.com/SajjadAemmi/SA-MobileNetV3/actions/workflows/python-package.yml/badge.svg)](https://github.com/SajjadAemmi/SA-MobileNetV3/actions/workflows/python-package.yml)

Shuffle Attention for MobileNetV3

![model arch](https://raw.githubusercontent.com/SajjadAemmi/SA-MobileNetV3/main/assets/model_arch.png)

## Reference
- SA-NET: Shuffle Attention for deep convolutional neural networks: https://github.com/wofmanaf/SA-Net

## Experiments

#### on ImageNet

Attempt | Parameters | Madds | Top1-acc | Sample visualization |
--- | --- | --- | --- | --- |
MobileNetV3-Large | 5.4 M | 448.69 M | 75.2%  | 
SA-MobileNetV3-Large | **3.9 M** | **445.68 M** | **76.8%** |

#### on CIFAR-10

Attempt | Parameters | Madds | Top1-acc | Sample visualization |
--- | --- | --- | --- | --- |
MobileNetV3-Large| 4.2 M | 446.16 M |  |
SA-MobileNetV3-Large | **2.7 M** | **443.14 M** |  |

#### on MNist

Attempt | Parameters | Madds | Top1-acc | Sample visualization | Sample visualization |
--- | --- | --- | --- | --- | --- |
MobileNetV3-Large | 4.2 M | 446.16 M | 0.997% | ![model arch](https://raw.githubusercontent.com/SajjadAemmi/SA-MobileNetV3/main/assets/mobilenetv3/7.jpg) | ![model arch](https://raw.githubusercontent.com/SajjadAemmi/SA-MobileNetV3/main/assets/mobilenetv3/7_2.jpg) ||
SA-MobileNetV3-Large | **2.7 M** | **443.14 M** | **0.998%** |![model arch](https://raw.githubusercontent.com/SajjadAemmi/SA-MobileNetV3/main/assets/sa-mobilenetv3/7.jpg) | ![model arch](https://raw.githubusercontent.com/SajjadAemmi/SA-MobileNetV3/main/assets/sa-mobilenetv3/7_2.jpg) ||

## Train

Run the following command for train model on your own dataset:
```
python train.py --dataset mnist 
```

## Test

Run the following command for evaluation trained model on test dataset:
```
python test.py --dataset mnist
```

## Inference

Run the following command for classification images:
```
python inference.py --input /path/to/image.jpg 
```

## Citation
Please cite our paper if you find this repo useful in your research.
```
@article{
}
```

SA-MobileNetV3

SA-MobileNetV3 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

SA-MobileNetV3

SA-MobileNetV3 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

Download README