Add VOC Detection and Segmentation Datasets #86

ellisbrown · 2017-03-05T20:31:57Z

took what @fmassa left off with at https://github.com/pytorch/examples/pull/21/files#diff-0344e770fabb635e92d12f8b67f3504a and completed / improved

VOCDetection should be complete. Should use AnnotationTransform when making the dataset and the detection_collate when making a DataLoader
VOCSegmentation not fully tested

torchvision/datasets/voc.py

+    def __getitem__(self, index):
+        img_id = self.ids[index]
+
+        target = Image.open(self._annopath % img_id).convert('RGB')


torchvision/datasets/voc.py

+    """VOC Segmentation Dataset Object
+    input and target are both images
+
+    NOTE: need to address https://github.com/pytorch/vision/issues/9


acgtyrant · 2017-07-04T09:53:30Z

When do you merge this pull request...

torchvision/datasets/voc.py

+            (default: alphabetic indexing of VOC's 20 classes)
+        keep_difficult (bool, optional): keep difficult instances or not
+            (default: False)
+        channels (int): number of channels


torchvision/datasets/voc.py

+            for i, cur_bb in enumerate(bbox):
+                bb_sz = int(cur_bb.text) - 1
+                # scale height or width
+                bb_sz = bb_sz / width if i % 2 == 0 else bb_sz / height


torchvision/datasets/voc.py

+    targets = []
+    imgs = []
+    for _, sample in enumerate(batch):
+        for _, tup in enumerate(sample):


torchvision/datasets/voc.py

+
+            label_ind = self.class_to_ind[name]
+            bndbox.append(label_ind)
+            res += [bndbox]  # [xmin, ymin, xmax, ymax, ind]


fmassa · 2018-12-06T14:20:07Z

Replaced by #663

Thanks for the PR!

fmassa self-assigned this Mar 7, 2017

fmassa reviewed Mar 7, 2017

View reviewed changes

torchvision/datasets/voc.py

def __getitem__(self, index):

img_id = self.ids[index]

target = Image.open(self._annopath % img_id).convert('RGB')

This comment was marked as off-topic.

Sign in to view

fmassa reviewed Mar 7, 2017

View reviewed changes

ellisbrown added 4 commits March 11, 2017 21:48

full voc detection support and initial voc segmentation support

8b1beb4

added VOC to readme

f74673b

added detection_collate docstring and refactored AnnotationTransform

c68423b

fixed flake8 errors

81e38d1

ellisbrown force-pushed the master branch from 771d937 to 81e38d1 Compare March 12, 2017 03:49

Update utils.py

6db2c3a

acgtyrant suggested changes Jul 4, 2017

View reviewed changes

acgtyrant reviewed Jul 5, 2017

View reviewed changes

torchvision/datasets/voc.py

label_ind = self.class_to_ind[name]

bndbox.append(label_ind)

res += [bndbox] # [xmin, ymin, xmax, ymax, ind]

This comment was marked as off-topic.

Sign in to view

bpinaya mentioned this pull request Nov 19, 2018

VOCSegmentation, VOCDetection, linting passing, examples. #663

Merged

fmassa closed this Dec 6, 2018

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Add VOC Detection and Segmentation Datasets #86

Add VOC Detection and Segmentation Datasets #86

ellisbrown commented Mar 5, 2017

This comment was marked as off-topic.

This comment was marked as off-topic.

This comment was marked as off-topic.

This comment was marked as off-topic.

This comment was marked as off-topic.

This comment was marked as off-topic.

acgtyrant commented Jul 4, 2017

This comment was marked as off-topic.

This comment was marked as off-topic.

This comment was marked as off-topic.

This comment was marked as off-topic.

This comment was marked as off-topic.

fmassa commented Dec 6, 2018

Add VOC Detection and Segmentation Datasets #86

Add VOC Detection and Segmentation Datasets #86

Conversation

ellisbrown commented Mar 5, 2017

This comment was marked as off-topic.

This comment was marked as off-topic.

This comment was marked as off-topic.

This comment was marked as off-topic.

This comment was marked as off-topic.

This comment was marked as off-topic.

acgtyrant commented Jul 4, 2017

This comment was marked as off-topic.

This comment was marked as off-topic.

This comment was marked as off-topic.

This comment was marked as off-topic.

This comment was marked as off-topic.

fmassa commented Dec 6, 2018