https://github.com/TannerGilbert/Object-Detection-and-Image-Segmentation-with-Detectron2
Object Detection and Image Segmentation with Detectron2
https://github.com/TannerGilbert/Object-Detection-and-Image-Segmentation-with-Detectron2
detectron2 detectron2-inference instance-segmentation object-detection
Last synced: 11 months ago
JSON representation
Object Detection and Image Segmentation with Detectron2
- Host: GitHub
- URL: https://github.com/TannerGilbert/Object-Detection-and-Image-Segmentation-with-Detectron2
- Owner: TannerGilbert
- License: mit
- Created: 2019-11-08T19:58:57.000Z (over 6 years ago)
- Default Branch: master
- Last Pushed: 2022-09-05T21:07:08.000Z (over 3 years ago)
- Last Synced: 2024-08-08T23:18:42.311Z (almost 2 years ago)
- Topics: detectron2, detectron2-inference, instance-segmentation, object-detection
- Language: Jupyter Notebook
- Homepage: https://gilberttanner.com/blog/detectron-2-object-detection-with-pytorch
- Size: 129 MB
- Stars: 154
- Watchers: 6
- Forks: 40
- Open Issues: 2
-
Metadata Files:
- Readme: README.md
- Funding: .github/FUNDING.yml
- License: LICENSE
Awesome Lists containing this project
README
# 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](https://github.com/facebookresearch/detectron2/blob/master/INSTALL.md).
## 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](https://github.com/facebookresearch/detectron2/tree/master/docker).
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](https://github.com/TannerGilbert/Object-Detection-and-Image-Segmentation-with-Detectron2/tree/master/docker).
## Inference with pre-trained model
* [Detectron2 Inference with pretrained model](Detectron2_inference_with_pre_trained_model.ipynb).
* [Detect from Webcam or Video](detect_from_webcam_or_video.py)
* [Detectron2 RestAPI](deploy/rest-api)
## Training on a custom dataset
* [Detectron2 train on a custom dataset](Detectron2_train_on_a_custom_dataset.ipynb).
* [Detectron2 train with data augmentation](Detectron2_Train_on_a_custom_dataset_with_data_augmentation.ipynb)
* [Detectron2 Chess Detection](Detectron2_Detect_Chess_Detection.ipynb)
* [Detectron2 Vehicle Detection](Detectron2_Vehicle_Detection.ipynb)
## D2Go
* [D2GO_Introduction](D2Go/D2GO_Introduction.ipynb)
* [D2Go_Train_Microcontroller_Detector](D2Go/D2Go_Train_Microcontroller_Detector.ipynb)
## Author
**Gilbert Tanner**