__init__.py 1.1 KB

12345678910111213141516171819202122232425
  1. from cvdatasets.dataset.mixins.chainer_mixins import IteratorMixin
  2. from cvdatasets.dataset.mixins.features import PreExtractedFeaturesMixin
  3. from cvdatasets.dataset.mixins.parts import BBCropMixin
  4. from cvdatasets.dataset.mixins.parts import BBoxMixin
  5. from cvdatasets.dataset.mixins.parts import CroppedPartMixin
  6. from cvdatasets.dataset.mixins.parts import MultiBoxMixin
  7. from cvdatasets.dataset.mixins.parts import PartCropMixin
  8. from cvdatasets.dataset.mixins.parts import PartMixin
  9. from cvdatasets.dataset.mixins.parts import PartRevealMixin
  10. from cvdatasets.dataset.mixins.parts import PartsInBBMixin
  11. from cvdatasets.dataset.mixins.parts import RandomBlackOutMixin
  12. from cvdatasets.dataset.mixins.parts import RevealedPartMixin
  13. from cvdatasets.dataset.mixins.parts import UniformPartMixin
  14. from cvdatasets.dataset.mixins.reading import AnnotationsReadMixin
  15. from cvdatasets.dataset.mixins.reading import ImageListReadingMixin
  16. class ImageWrapperDataset(PartMixin, PreExtractedFeaturesMixin, AnnotationsReadMixin, IteratorMixin):
  17. pass
  18. class Dataset(ImageWrapperDataset):
  19. def get_example(self, i):
  20. im_obj = super(Dataset, self).get_example(i)
  21. return im_obj.as_tuple()