MobileNetV3_Small

Description

MobileNetV3_Small is a lightweight convolutional neural network architecture designed for efficient mobile and embedded devices. It is part of the MobileNet family, renowned for its compact size and high performance, making it ideal for applications with limited computational resources.The key focus of MobileNetV3_Small is to achieve a balance between model size, speed, and accuracy.

Setup

Install

pip3 install onnx
pip3 install tqdm

Download

Pretrained model: https://download.pytorch.org/models/mobilenet_v3_small-047dcff4.pth

Dataset: https://www.image-net.org/download.php to download the validation dataset.

Model Conversion

python3 export.py --weight mobilenet_v3_small-047dcff4.pth --output mobilenetv3_small.onnx

Inference

export DATASETS_DIR=/Path/to/imagenet_val/

FP16

# Accuracy
bash scripts/infer_mobilenetv3_small_fp16_accuracy.sh
# Performance
bash scripts/infer_mobilenetv3_small_fp16_performance.sh

Results