Open Source Computer Vision Classification Models
The Roboflow Model Library contains pre-configured model architectures for easily training computer vision models. Just add the link from your Roboflow dataset and you're ready to go! We even include the code to export to common inference formats like TFLite, ONNX, and CoreML.
If you'd like to request a model we haven't yet implemented, please get in touch.
Tensorflow 1.5 Classification
EfficientNet is from a family of image classification models from GoogleAI that train comparatively quickly on small amounts of data, making the most of limited datasets.
Tensorflow 2 Classification
MobileNet is a GoogleAI model well-suited for on-device, real-time classification (distinct from MobileNetSSD, Single Shot Detector). This implementation leverages transfer learning from ImageNet to your dataset.