1. 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
  2. 11 Jan, 2021 1 commit
    • Josh Bradley's avatar
      Add widerface dataset (#2883) · d0063f3d
      Josh Bradley authored
      
      
      * initial commit of widerface dataset
      
      * comment out old code
      
      * improve parsing of annotation files
      
      * code cleanup and fix docstring comments
      
      * speed up check for quota exceeded
      
      * cleanup print statements
      
      * reformat code and remove print statements
      
      * minor code cleanup and reformatting
      
      * add more comments
      
      * reuse variable
      
      * reverse formatting changes
      
      * fix flake8 errors
      
      * add type annotations
      
      * fix mypy errors
      
      * add a base_folder to root directory
      
      * some formatting fixes
      
      * GDrive threshold does not throw 403 error
      
      * testing new download logic
      
      * cleanup logic for download and integrity check
      
      * use a better variable name
      
      * format fix
      
      * reorder list in docstring
      
      * initial widerface unit test - fails on MD5 check
      
      * use list of dictionaries to store dataset
      
      * fix docstring formatting
      
      * remove unnecessary error checking
      
      * fix type checker error
      
      * revert typo fix
      
      * rename var constants, use file context manager, verify str args
      
      * fix flake8 error
      
      * fix checking target_type argument values
      
      * create uncompressed dataset folders
      
      * cleanup unit tests for widerface
      
      * use correct os function
      
      * add more info to docstring
      
      * disable unittests for windows
      
      * fix _check_integrity logic
      
      * update docstring
      
      * remove citation
      
      * remove target_type option
      
      * fix formatting issue
      Co-authored-by: default avatarPhilip Meier <github.pmeier@posteo.de>
      
      * remove comment and add more info to docstring
      
      * update type annotations
      
      * restart CI jobs
      Co-authored-by: default avatarJoshua Bradley <jgbrad3@evoforge.org>
      Co-authored-by: default avatarPhilip Meier <github.pmeier@posteo.de>
      Co-authored-by: default avatarvfdev <vfdev.5@gmail.com>
      d0063f3d
  3. 25 Aug, 2020 1 commit
    • 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
  4. 31 Jul, 2019 1 commit
  5. 24 Jul, 2019 1 commit
  6. 19 Jul, 2019 1 commit
    • Francisco Massa's avatar
      Add VideoClips and Kinetics dataset (#1077) · 5d1372c0
      Francisco Massa authored
      * Add VideoClips and Kinetics dataset
      
      * Lint + add back missing line
      
      * Adds ClipSampler following Bruno comment
      
      * Change name following Bruno's suggestion
      
      * Enable specifying a target framerate
      
      * Fix test_io for new interface
      
      * Add comment mentioning drop_last behavior
      
      * Make compute_clips more robust
      
      * Flake8
      
      * Fix for Python2
      5d1372c0
  7. 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
  8. 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
  9. 30 Apr, 2019 1 commit
  10. 29 Apr, 2019 1 commit
  11. 26 Mar, 2019 1 commit
  12. 19 Mar, 2019 1 commit
    • Philip Meier's avatar
      ImageNet dataset (#764) · 69382912
      Philip Meier authored
      * initial commit
      
      * fixed Python2 issue
      
      * fixed naming incorrectness and Python2 compability
      
      * fixed preparation of train folder
      
      * removed detection dataset
      
      * added docstring and repr
      
      * moved import of scipy to make the import of torchvision independent of it
      
      * improved conversion from class string to index
      
      * removed support for other years than 2012
      
      * removed accidentally added file
      
      * moved emptying of split folder to avoid accidental deletion
      
      * removed deletion of the images
      
      * removed error conversion for Python2
      
      * Aligned class indices with the indices identified by ImageFolder class
      69382912
  13. 07 Mar, 2019 1 commit
    • Bryan He's avatar
      Add Caltech101, Caltech256, and CelebA (#775) · 9d9f48a3
      Bryan He authored
      * Add Caltech101 and Caltech256
      
      * Add information about default for target_type
      
      * Fix docs
      
      * Add function to download from Google Drive
      
      * Add CelebA dataset
      
      * Only import pandas when needed
      
      * Addressing comments
      
      * Remove trailing whitespace
      
      * Replace torch.LongTensor with torch.as_tensor
      9d9f48a3
  14. 25 Dec, 2018 1 commit
  15. 21 Dec, 2018 1 commit
  16. 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
  17. 04 Dec, 2018 2 commits
  18. 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
  19. 09 Mar, 2018 2 commits
  20. 28 Jan, 2018 1 commit
    • Sanyam Kapoor's avatar
      Omniglot Dataset (#323) · dac9efae
      Sanyam Kapoor authored
      * Add basic Omniglot dataset loader
      
      * Remove unused import
      
      * Add Omniglot random pair to sample pair of characters
      
      * Precompute random set of pairs, deterministic after object instantiation
      
      * Export OmniglotRandomPair via the datasets module interfact
      
      * Fix naming convention, use sum instead of reduce
      
      * Fix downloading to not download everything, fix Python2 syntax
      
      * Fix end line lint
      
      * Add random_seed, syntax fixes
      
      * Remove randomized pair, take up as a separate generic wrapper
      
      * Fix master conflict
      dac9efae
  21. 06 Dec, 2017 1 commit
  22. 07 Nov, 2017 1 commit
  23. 02 Sep, 2017 1 commit
  24. 20 Jul, 2017 1 commit
  25. 18 Mar, 2017 2 commits
  26. 16 Mar, 2017 1 commit
  27. 02 Mar, 2017 1 commit
  28. 17 Jan, 2017 1 commit
  29. 10 Nov, 2016 4 commits