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 B2

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

MobileNetV2 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.

Fast.ai v2 Classification

ResNet-32

A fast, simple convolutional neural network that gets the job done for many tasks, including classification here. Read More...