# SA-MobileNetV3

**Repository Path**: chaucerg/SA-MobileNetV3

## Basic Information

- **Project Name**: SA-MobileNetV3
- **Description**: Shuffle Attention for MobileNetV3
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 1
- **Forks**: 0
- **Created**: 2021-11-13
- **Last Updated**: 2023-02-15

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# SA-MobileNetV3
Shuffle Attention for MobileNetV3

![model arch](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](assets/mobilenetv3/7.jpg) | ![model arch](assets/mobilenetv3/7_2.jpg) ||
SA-MobileNetV3-Large | **2.7 M** | **443.14 M** | **0.998%** |![model arch](assets/sa-mobilenetv3/7.jpg) | ![model arch](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
```

## 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{
}
```