1. 10 Jan, 2022 1 commit
    • puhuk's avatar
      Add Country dataset (#5138) · ffd0d237
      puhuk authored
      
      
      * Add Country211 dataset
      
      To addresses issue #5108.
      
      * Add Country211 dataset
      
      To addresses issue #5108.
      
      * Update country211.py
      
      * Update country211.py
      
      * Code review reflected
      
      Reflect code review
      
      * Update test_datasets.py
      
      * Update with review
      
      Update with review
      
      * inherit from ImageFolder
      
      * Update test/test_datasets.py
      
      * Docstring + minor test update
      Co-authored-by: default avatarPhilip Meier <github.pmeier@posteo.de>
      Co-authored-by: default avatarNicolas Hug <nicolashug@fb.com>
      ffd0d237
  2. 07 Jan, 2022 1 commit
  3. 06 Jan, 2022 2 commits
    • Philip Meier's avatar
      OxfordIIITPet dataset (#5116) · 058f4bd7
      Philip Meier authored
      * add prototype dataset for oxford-iiit-pet
      
      * add old-style dataset
      
      * add tests
      
      * fix mypy
      
      * fix test
      
      * remove properties and use pathlib
      
      * target_type to target_types
      
      * move target annotation
      
      * add docstring
      
      * fix test
      058f4bd7
    • Sumukh Aithal's avatar
      Add GTSRB dataset (#5117) · 7120024d
      Sumukh Aithal authored
      
      
      * Added GTSRB dataset
      
      * Added unittest for GTSRB dataset
      
      * Apply suggestions from code review
      
      * More changes from code review
      
      * readd accidental removed line
      Co-authored-by: default avatarPhilip Meier <github.pmeier@posteo.de>
      7120024d
  4. 05 Jan, 2022 2 commits
  5. 22 Dec, 2021 1 commit
    • Joao Gomes's avatar
      Food 101 dataset (#5119) · 65676b4b
      Joao Gomes authored
      
      
      * Adding multiweight support for shufflenetv2 prototype models
      
      * Revert "Adding multiweight support for shufflenetv2 prototype models"
      
      This reverts commit 31fadbee7d1a65cd73ae43dfd4ac6e97e7ca7b01.
      
      * Adding multiweight support for shufflenetv2 prototype models
      
      * Revert "Adding multiweight support for shufflenetv2 prototype models"
      
      This reverts commit 4e3d900f796c1e3e667312087e77956ca4a4c017.
      
      * Add Food101 Dataset
      
      Addresses #5108.
      cc @pmeier @NicolasHug
      
      * Remove unecessary Path contructor calls
      
      * Remove unecessary Path contructor callsi and fix types
      
      * Fix tests
      
      * Address PR comments from @pmeier
      
      * Fix bug in tests and in food101 dataset
      
      * Fix bug in tests and in food101 dataset
      
      * Update torchvision/datasets/food101.py
      Co-authored-by: default avatarPhilip Meier <github.pmeier@posteo.de>
      65676b4b
  6. 12 Nov, 2021 1 commit
  7. 05 Nov, 2021 2 commits
  8. 04 Nov, 2021 1 commit
  9. 01 Nov, 2021 1 commit
  10. 14 Sep, 2021 1 commit
    • Muhammed Abdullah's avatar
      Added LFW Dataset (#4255) · d85aa6d3
      Muhammed Abdullah authored
      
      
      * Added LFW Dataset
      
      * Added dataset to list in __init__.py
      
      * Updated lfw.py
      * Created a common superclass for people and pairs type datatsets
      * corrected the .download() method
      
      * Added docstrings and updated datasets.rst
      
      * Wrote tests for LFWPeople and LFWPairs
      
      * Resolved mypy error: Need type annotation for "data"
      
      * Updated inject_fake_data method for LFWPeople
      
      * Updated tests for LFW
      
      * Updated LFW tests and minor changes in lfw.py
      
      * Updated LFW
      * Added functionality for 10-fold validation view
      * Optimized the code so to replace repeated lines by method in super
        class
      * Updated LFWPeople to get classes from all lfw-names.txt rather than
        just the classes fron trainset
      
      * Updated lfw.py and tests
      * Updated inject_fake_data method to create 10fold fake data
      * Minor changes in docstring and extra_repr
      
      * resolved py lint errors
      
      * Added checksums for annotation files
      
      * Minor changes in test
      
      * Updated docstrings, defaults and minor changes in test
      
      * Removed 'os.path.exists' check
      Co-authored-by: default avatarABD-01 <myac931@gmai.com>
      Co-authored-by: default avatarPhilip Meier <github.pmeier@posteo.de>
      Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
      d85aa6d3
  11. 01 Jul, 2021 1 commit
  12. 04 Jun, 2021 1 commit
  13. 28 Apr, 2021 1 commit
  14. 09 Apr, 2021 1 commit
    • Prabhat Roy's avatar
      Added KITTI dataset (#3640) · 7da9afee
      Prabhat Roy authored
      
      
      * Added KITTI dataset
      
      * Addressed review comments
      
      * Changed type of target to List[Dict] and corrected the data types of the returned values.
      
      * Updated unit test to rely on ImageDatasetTestCase
      
      * Added kitti to dataset documentation
      
      * Cleaned up test and some minor changes
      
      * Made data_url a string instead of a list
      
      * Removed unnecessary try and print
      Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
      7da9afee
  15. 12 Mar, 2021 1 commit
  16. 26 Jan, 2021 1 commit
  17. 04 Jan, 2021 1 commit
  18. 25 Aug, 2020 2 commits
    • Philip Meier's avatar
      Places365 dataset (#2610) · fc69c225
      Philip Meier authored
      * initial draft
      
      * [dirty] progress
      
      * remove inheritance from ImageFolder
      
      * add tests
      
      * lint
      
      * fix type hints
      
      * align getitem with other datasets
      
      * remove unused import
      
      * add docstring
      
      * guard existing image folders from overwrite
      
      * add missing entry in docstring
      
      * make fixpath more legible
      
      * add Places365 to docs
      fc69c225
    • Marijan Smetko's avatar
      3e0f5a6f
  19. 03 Aug, 2020 1 commit
  20. 15 Apr, 2020 1 commit
  21. 01 Aug, 2019 1 commit
  22. 31 Jul, 2019 1 commit
  23. 12 Jun, 2019 1 commit
    • Leon Bottou's avatar
      Added support for the QMNIST dataset (#995) · ac2e995a
      Leon Bottou authored
      * Added general reader for sn3 tensors in "pascalvincent" format
      
      * Added class QMNIST into mnist.py
      
      * QMNIST dataset: make some pt files smaller
      
      * Change request from fmassa.
      
      * read_sn3_pascalvincent_tensor: cse
      
      * read_sn3_pascalvincent_tensor: check file size (when strict!=False)
      
      * Fix lint
      
      * More lint
      
      * Add documentation and expose QMNIST to dataset namespace
      ac2e995a
  24. 27 May, 2019 1 commit
    • Francisco Massa's avatar
      Usps dataset (#961) · d4a126b6
      Francisco Massa authored
      * add USPS dataset
      
      * minor fixes
      
      * Improvements to the USPS dataset
      
      Add it to the documentation, expose it to torchvision.datasets
      and inherit from VisionDataset
      d4a126b6
  25. 22 May, 2019 1 commit
  26. 18 Apr, 2019 1 commit
  27. 26 Mar, 2019 1 commit
  28. 23 Feb, 2019 1 commit
  29. 25 Dec, 2018 1 commit
  30. 21 Dec, 2018 1 commit
  31. 06 Dec, 2018 1 commit
    • Benjamin Pinaya's avatar
      VOCSegmentation, VOCDetection, linting passing, examples. (#663) · e489abc6
      Benjamin Pinaya authored
      
      
      * VOC Dataset, linted, flak8 passing, samples on gist.
      
      * Double backtick on values.
      
      * Apply suggestions from code review
      
      Add suggestions from @ellisbrown, using dict of dicts instead of array index.
      Co-Authored-By: default avatarbpinaya <bpg_92@hotmail.com>
      
      * Fixed errors with the new comments.
      
      * Added documentation on RST
      
      * Removed getBB, added parse_voc_xml, variable naming change.
      
      * Removed unused variable, removed VOC_CLASSES, two new gists for test.
      e489abc6
  32. 04 Dec, 2018 2 commits
  33. 10 Mar, 2018 1 commit
    • Frédérik Paradis's avatar
      Adding a DatasetFolder class. (#444) · fa377c47
      Frédérik Paradis authored
      * Adding tests to ImageFolder
      
      * Adding DatasetFolder class
      
      * Fix tests for pytest and code for lint checker
      
      * Adding mock to requirements for ImageFolder tests
      
      * Remove mocks from requirements
      fa377c47
  34. 09 Mar, 2018 2 commits