apple/ml-cvnets

bug in coco_ssd.py

MLRadfys opened this issue · 2 comments

MLRadfys commented

Hi,

I noticed a small mistake in the coco_ssd.py script.

In def getitem(...):

boxes, labels = self.get_boxes_and_labels( image_id=image_id, image_width=im_width, image_height=im_height )

self.get_boxes_and_labels() returns boxes, labels and masks (or None) and should be changed to:

boxes, labels, _ = self.get_boxes_and_labels( image_id=image_id, image_width=im_width, image_height=im_height )

Cheers,

M

darwinharianto commented

ml-cvnets/data/datasets/detection/coco_ssd.py

Line 158 in c668ad1

boxes, labels, _ = self.get_boxes_and_labels(

it looks correct

mchorton commented

Marking resolved per @darwinharianto’s comment