MobileNetV3 in Python | Shuffle Attention for MobileNetV3 in Python

SA-MobileNetV3

Shuffle Attention for MobileNetV3

Reference

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

Experiments

on ImageNet

on CIFAR-10

on MNist

Attempt	Parameters	Madds	Top1-acc	Sample visualization	Sample visualization
MobileNetV3-Large	4.2 M	446.16 M	0.997%
SA-MobileNetV3-Large	2.7 M	443.14 M	0.998%

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

Predict

Run the following command for classification images:

python predict.py --input /path/to/image.jpg 

Citation

Please cite our paper if you find this repo useful in your research.

@article{
}

Download Details: 
 

Author: SajjadAemmi

Official Website: https://github.com/SajjadAemmi/SA-MobileNetV3