TannerGilbert / Object-Detection-and-Image-Segmentation-with-Detectron2

Object Detection and Image Segmentation with Detectron2

Home Page:https://gilberttanner.com/blog/detectron-2-object-detection-with-pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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.

Installation

See the official installation guide.

Install using Docker

Another great way to install Detectron2 is by using Docker. Docker is great because you don't need to install anything locally, which allows you to keep your machine nice and clean.

If you want to run Detectron2 with Docker you can find a Dockerfile and docker-compose.yml file in the docker directory of the repository.

For those of you who also want to use Jupyter notebooks inside their container, I created a custom Docker configuration, which automatically starts Jupyter after running the container. If you're interested you can find the files in the docker directory.

Inference with pre-trained model

Training on a custom dataset

D2Go

Author

Gilbert Tanner

About

Object Detection and Image Segmentation with Detectron2

https://gilberttanner.com/blog/detectron-2-object-detection-with-pytorch

License:MIT License


Languages

Language:Jupyter Notebook 100.0%Language:Python 0.0%Language:Dockerfile 0.0%