/caffe-mobilenet-v3

Caffe Implementation of MobileNets V3

MIT LicenseMIT

caffe-mobilenet-v3

Introduction

This is a personal Caffe implementation of MobileNetV3. For details, please read the original papers: Searching for MobileNetV3.

How to use

  1. Requirements for Caffe (see: Caffe installation instructions)
  2. Add new caffe layers and rebuild the caffe:
  3. Run test
    CPU:
    $CAFFE_ROOT/build/tools/caffe time -model mobilenet_v3_large_1.0.prototxt
    GPU:
    $CAFFE_ROOT/build/tools/caffe time -model mobilenet_v3_large_1.0.prototxt -gpu 0

Performance on MobileNetV3

Backbone CPU-Forward CPU-Backward GPU-Forward GPU-Backward
V3-Large 1.0 134.55 ms 140.23 ms 15.44 ms 21.79 ms
V3-Small 1.0 58.64 ms 59.30 ms 11.49 ms 12.58 ms

TODO

  • More MobileNetV3 architectures.
  • Traning & validation.