Skip to content
Object Detection and Image Segmentation with Detectron2
Jupyter Notebook
Branch: master
Clone or download
Type Name Latest commit message Commit time
Failed to load latest commit information.
.gitignore fixed variable name bug Nov 26, 2019
Detectron2_inference_with_pre_trained_model.ipynb added webcam videostream detection and fixed .imshow error Nov 9, 2019
Detectron2_train_on_a_custom_dataset.ipynb added webcam videostream detection and fixed .imshow error Nov 9, 2019
LICENSE train and run inference with detectron2 Nov 8, 2019 updated ReadME Nov 8, 2019

Object Detection and Instance Segmentation with Detecton2

Detectron2 is Facebooks new library that implements state-of-the-art object detection algorithm. This repository shows you how to use Detectron2 for both inference as well as using transfer learning to train on your own data-set.


See the official installation guide.

Inference with pre-trained model

Using a pre-trained model for inference is as easy as loading in the cofiguration and weights and creating a predictor object. For a example check out Detectron2_inference_with_pre_trained_model.ipynb.

Training on a custom dataset

Training a model on a custom dataset is a bit more challenging because it requires that we register our dataset. For a example showing you how to train a balloon detection model check out Detectron2_train_on_a_custom_dataset.ipynb.


Gilbert Tanner

You can’t perform that action at this time.