Written by

Paul Guerrie

Paul Guerrie

Machine Learning Engineer at Roboflow - Building tools to democratize computer vision.

How to Train YOLOv5 Instance Segmentation on a Custom Dataset

YOLOv5 is usually associated with object detection and is one of the most popular networks in the world for that task. Recently, image classification was added to YOLOv5, and it

How to Train YOLOv5-Classification on a Custom Dataset

YOLOv5 is one of the most popular object detection networks in the world, and now object detection isn't the only trick up its sleeve! As of August 2022, YOLOv5 also

How To Train SegFormer on a Custom Dataset

In this post, we will walk through how to train SegFormer on a custom dataset using Pytorch Lightning to classify every pixel in an image.

How to Train Detectron2 for Custom Instance Segmentation

A walk through on how to train Detectron2 to segment your custom objects from any image by providing our model with example training data.

What to Think About When Choosing Model Sizes

When training any machine learning model, you must trade off inference speed for accuracy. Larger models with more parameters are uniformly more accurate, and smaller models with fewer parameters are uniformly faster to infer.