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.

Fast.ai v2 Classification

Resnet34

Resnet34 for state of the art image classification implemented in fastai v2 and PyTorch Read More...

Keras Classification

EfficientNet

EfficientNet is a family of state of the art classification models from GoogleAI that efficiently scale up as you increase the number of parameters in the network. Read More...

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