/notebooks

Examples and tutorials on using SOTA computer vision models and techniques. Learn everything from old-school ResNet, through YOLO and object-detection transformers like DETR, to the latest models like Grounding DINO and SAM.

Primary LanguageJupyter Notebook


👋 hello

Over the years we have created dozens of Computer Vision tutorials. This repository contains Jupyter Notebooks that we've featured in our blog posts and YouTube videos. Please keep in mind that repository is still under construction 🚧 In the meantime, you can find a full list of our tutorials here.

🚀 model tutorials

notebook open in colab / kaggle / sagemaker studio lab complementary materials repository / paper
Segment Anything Model Colab Kaggle SageMaker Roboflow YouTube GitHub arXiv
Zero-Shot Object Detection with Grounding DINO Colab Kaggle SageMaker Roboflow YouTube GitHub arXiv
DETR Transformer PyTorch Object Detection Colab Kaggle SageMaker YouTube GitHub arXiv
YOLOv8 PyTorch Object Detection Colab Kaggle SageMaker Roboflow YouTube GitHub
YOLOv8 PyTorch Instance Segmentation Colab Kaggle SageMaker YouTube GitHub
YOLOv8 PyTorch Classification Colab Kaggle SageMaker GitHub
YOLOv7 PyTorch Object Detection Colab Kaggle Roboflow YouTube GitHub arXiv
YOLOv7 PyTorch Instance Segmentation Colab Kaggle SageMaker Roboflow YouTube GitHub arXiv
YOLOv7 PyTorch Object Detection OpenVINO + TorchORT Colab Kaggle Roboflow GitHub arXiv
MT-YOLOv6 PyTorch Object Detection Colab Kaggle Roboflow YouTube GitHub arXiv
YOLOv5 PyTorch Object Detection Colab Kaggle Roboflow YouTube GitHub
YOLOv5 PyTorch Classification Colab Kaggle Roboflow YouTube GitHub
YOLOv5 PyTorch Instance Segmentation Colab Kaggle SageMaker Roboflow YouTube GitHub
Detection2 PyTorch Instance Segmentation Colab Kaggle Roboflow YouTube GitHub arXiv
SegFormer PyTorch Instance Segmentation Colab Kaggle Roboflow YouTube GitHub arXiv
Vision Transformer PyTorch Classification Colab Kaggle Roboflow YouTube GitHub arXiv
Scaled-YOLOv4 PyTorch Object Detection Colab Kaggle Roboflow YouTube GitHub arXiv
YOLOS PyTorch Object Detection Colab Kaggle Roboflow YouTube GitHub arXiv
YOLOR PyTorch Object Detection Colab Kaggle Roboflow YouTube GitHub arXiv
YOLOX PyTorch Object Detection Colab Kaggle Roboflow YouTube GitHub arXiv
Resnet34 fast.ai Classification Colab Kaggle Roboflow YouTube
OpenAI Clip PyTorch Classification Colab Kaggle Roboflow YouTube GitHub
YOLOv4-tiny Darknet Object Detection Colab Kaggle Roboflow YouTube GitHub arXiv

📸 computer vision skills

notebook open in colab / kaggle / sagemaker studio lab complementary materials repository / paper
Automated Dataset Annotation and Evaluation with Grounding DINO and SAM Colab Kaggle SageMaker YouTube GitHub arXiv
Automated Dataset Annotation and Evaluation with Grounding DINO Colab Kaggle SageMaker YouTube GitHub arXiv
Vector Analysis with Scikit-learn and Bokeh Colab Kaggle Roboflow
RF100 Object Detection Model Benchmarking Colab Kaggle Roboflow YouTube GitHub arXiv
Detect and Count Objects in Polygon Zone with YOLOv5 / YOLOv8 / Detectron2 + Supervision Colab Kaggle SageMaker YouTube GitHub
Track and Count Vehicles with YOLOv8 + ByteTRACK + Supervision Colab Kaggle SageMaker Roboflow YouTube GitHub arXiv
Football Players Tracking with YOLOv5 + ByteTRACK Colab Kaggle SageMaker Roboflow YouTube GitHub arXiv
Create Segmentation Masks with Roboflow Colab Kaggle Roboflow
How to Use PolygonZone and Roboflow Supervision Colab Kaggle Roboflow

🎬 videos

Almost every week we create tutorials showing you the hottest models in Computer Vision. 🔥 Subscribe, and stay up to date with our latest YouTube videos!

automated-dataset-annotation-and-evaluation-with-grounding-dino-and-sam

segment-anything-model-tutorial

💻 run locally

We try to make it as easy as possible to run Roboflow Notebooks in Colab and Kaggle, but if you still want to run them locally, below you will find instructions on how to do it. Remember don't install your dependencies globally, use venv.

# clone repository and navigate to root directory
git clone git@github.com:roboflow-ai/notebooks.git
cd notebooks

# setup python environment and activate it
python3 -m venv venv
source venv/bin/activate

# install and run jupyter notebook
pip install notebook
jupyter notebook

☁️ run in sagemaker studio lab

You can now open our tutorial notebooks in Amazon SageMaker Studio Lab - a free machine learning development environment that provides the compute, storage, and security—all at no cost—for anyone to learn and experiment with ML.

Stable Diffusion Image Generation YOLOv5 Custom Dataset Training YOLOv7 Custom Dataset Training
SageMaker SageMaker SageMaker

🐞 bugs & 🦸 contribution

Computer Vision moves fast! Sometimes our notebooks lag a tad behind the ever-pushing forward libraries. If you notice that any of the notebooks is not working properly, create a bug report and let us know.

If you have an idea for a new tutorial we should do, create a feature request. We are constantly looking for new ideas. If you feel up to the task and want to create a tutorial yourself, please take a peek at our contribution guide. There you can find all the information you need.

We are here for you, so don't hesitate to reach out.