f8aeed8e2ba799fd4d964e46479acfd5c768716c,luminoth/tools/dataset/readers/object_detection/imagenet.py,ImageNetReader,iterate,#ImageNetReader#,48

Before Change

            self.yielded_records += 1

            yield {
                "width": width,
                "height": height,
                "depth": 3,
                "filename": annotation["filename"],
                "image_raw": image,
                "gt_boxes": gt_boxes,
            }

    def _validate_structure(self):
        if not tf.gfile.Exists(self._data_dir):
            raise InvalidDataDirectory(

After Change

            if len(gt_boxes) == 0:
                continue

            record = {
                "width": width,
                "height": height,
                "depth": 3,
                "filename": annotation["filename"],
                "image_raw": image,
                "gt_boxes": gt_boxes,
            }

            self._will_add_record(record)
            self.yielded_records += 1

            yield record

    def _validate_structure(self):
        if not tf.gfile.Exists(self._data_dir):
            raise InvalidDataDirectory(

Instances