Train
Our managed computer vision training solution will give you a state of the art model hosted at an API endpoint customized for your dataset in no time.
Roboflow offers an AutoML product called Roboflow Train. It is the easiest way to train and deploy a state of the art object detection model on your custom dataset. It's literally one click -- we'll do the rest. When your model is done training, you'll receive access to a hosted inference API to interrogate your model for predictions via your programming language of choice (or a simple demo web app), a Tensorflow JS model you can embed in your web application, and an on-device inference server you can run on edge devices like the NVIDIA Jetson.
Roboflow Train credits are available in your workspace depending on which plan you have selected for the workspace. Training a model costs one credit and takes between 1 and 24 hours depending on the size of your dataset. Contact our sales team to upgrade your plan if you need more train credits.
- Contribute to the Roboflow Blog: Author a post, and share your work with Roboflow blog and newsletter readers.
- Apply for Research Credits: Additional training credits and increased account limits are available for research and education.
Choose image preprocessing and image augmentation settings, then generate a version of your dataset. Click "Use Roboflow Train" then "Start Training" and we will train a model for you and return back an API you can use.
- The default image resize is 416x416. If you select a "Resize" preprocessing option, we will train a model whose native input size is similar to your training data size.
- To make your model faster, try exporting images in a smaller size.
The maximum size we support is 1024x1024 -- which is quite slow to train and conduct inference against. If you're not sure what to pick, we recommend starting with 416x416.
The larger your dataset and the larger your outputted images, the longer it will take for your dataset to train. We will email you when it's finished. In most cases, this should be under 24 hours.
You can monitor your training job's progress. The UI will show your machine starting up to begin training:
Roboflow Train: Training Machine Starting
When your model has finished training, you can see the metrics on the dataset Versions page including mean Average Precision (mAP), precision, recall, and more.
Model Training Results: Roboflow Train
Model Training Graphs: Roboflow Train
- You can save the training graph by right-clicking it and selecting "Save Image As."
You can also see how your model performs on images from your dataset and local and hosted image and video files with the Deploy Tab.
The Roboflow Deploy tab
You can also see how your model performed on the specific images in your test and validation sets by using the "Visualize" feature to get a subjective feel for what these training metrics mean for your particular dataset.
After training, your model is ready to be used for inference and embedded in a custom application! See the Inference Documentation page for all the options.
Last modified 4h ago