Skip to content

sirius-ai/MobileNetV3-TF

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

MobileNetV3-TF

Tensorflow implementation for two new MobileNetV3 models, these models can be use in classification、detection and segmentation task at so on.
For detail info please reference the paper Searching for MobileNetV3.

Mobilenet V3 block

MobileNetV3-block

MobileNetV3-Small-Spec

MobileNetV3-Small-Spec

MobileNetV3-Large-Spec

MobileNetV3-Large-Spec

dependencies

  • tensorflow >= r1.14
  • python 3.6.x
  • numpy

prepare datasets

  1. firstly, structure your datasets like below:
    datasets directory struct
  2. secondly, convert your datasets to tfrecords format.

training

when well done config the arguments run 'python train_nets.py'

testing

when well done config the arguments run 'python test_nets.py'

References

  1. Searching for MobileNetV3
  2. TensorFlow Core r2.0

Releases

No releases published

Packages

No packages published

Languages