Skip to content
This repository has been archived by the owner on Jan 26, 2022. It is now read-only.

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

Open
wants to merge 55 commits into
base: master
Choose a base branch
from

Conversation

adityaarun1
Copy link

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
Copy link

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
Copy link

Many thanks for your work! @adityaarun1 @ruotianluo

@adityaarun1 ‘s fork is great!

BTW, is this repo still alive?

@adityaarun1
Copy link
Author

adityaarun1 commented May 30, 2019

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

@adityaarun1 adityaarun1 changed the title Support for PyTorch 1.0, PASCAL VOC Dataset and Custom Dataset Support for PyTorch 1.0, TorchVision 0.3.0, PASCAL VOC Dataset and Custom Dataset May 30, 2019
Fixes #10
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants