__init__.py 498 B

1234567891011121314
  1. from .mixins.reading import AnnotationsReadMixin, ImageListReadingMixin
  2. from .mixins.parts import PartMixin, RevealedPartMixin, CroppedPartMixin
  3. from .mixins.features import PreExtractedFeaturesMixin
  4. from .mixins.chainer_mixins import IteratorMixin
  5. class ImageWrapperDataset(PartMixin, PreExtractedFeaturesMixin, AnnotationsReadMixin, IteratorMixin):
  6. pass
  7. class Dataset(ImageWrapperDataset):
  8. def get_example(self, i):
  9. im_obj = super(Dataset, self).get_example(i)
  10. return im_obj.as_tuple()