__init__.py 434 B

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