Skip to content

Latest commit

 

History

History
31 lines (23 loc) · 1.24 KB

README.md

File metadata and controls

31 lines (23 loc) · 1.24 KB

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.