roytseng-tw/Detectron.pytorch
A PyTorch implementation of Facebook's Detectron for training and deploying object detection, instance segmentation, and keypoint estimation models.

This repository provides a pure PyTorch implementation of Facebook Research’s Detectron framework, enabling object detection, instance segmentation, and human pose estimation. It supports training models from scratch using custom datasets and inference using pretrained Detectron weight files. The implementation includes support for multi-GPU training, memory-efficient batching techniques, and various ROI pooling methods aligned with the original Caffe2-based Detectron.