Detectron.pytorch icon indicating copy to clipboard operation
Detectron.pytorch copied to clipboard

Published 20 hours ago • verified publisher icon roytseng-tw

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

Open adityaarun1 opened this issue 6 years ago • 3 comments

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.

adityaarun1 avatar Jan 17 '19 09:01 adityaarun1

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

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

ruotianluo avatar Feb 10 '19 04:02 ruotianluo

Many thanks for your work! @adityaarun1 @ruotianluo

@adityaarun1 ‘s fork is great!

BTW, is this repo still alive?

Hanqing-Sun avatar Mar 09 '19 02:03 Hanqing-Sun

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

adityaarun1 avatar May 30 '19 07:05 adityaarun1