search

roytseng-tw / Detectron.pytorch

A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.
MIT License
2.82k stars 567 forks source link

Support for PyTorch 1.0, TorchVision 0.3.0, PASCAL VOC Dataset and Custom Dataset #194

Open adityaarun1 opened 5 years ago

adityaarun1 commented 5 years ago

Added support for:

  1. PyTorch 1.0: Borrowed code from facebookresearch/maskrcnn-benchmark for ROIPool, ROIAlign and NMS.
  2. Pascal VOC 2007 and 2012: Followed the implementation of facebookresearch/Detectron to add support for VOC 2007 & 2012 data sets. Added tools/pascal_voc_xml2json.py file to convert VOC styled annotation to JSON format.
  3. Custom Dataset: Following VOC data set, added support to train custom data set.
ruotianluo commented 5 years ago

I squashed the pytorch 1.0 related commits to a single commit.

https://gist.github.com/ruotianluo/582c0e9b46ea202ed96eaddf08f80034#file-detectron-pytorch_pth1_patch-diff

Hanqing-Sun commented 5 years ago

Many thanks for your work! @adityaarun1 @ruotianluo

@adityaarun1 ‘s fork is great!

BTW, is this repo still alive?

adityaarun1 commented 5 years ago

Updated the code base for torchvision 0.3.0. Now the compilation of various ROI layers is not required.