Skip to content

GitLab

Switch branch/tag
  1. 11 Apr, 2022 1 commit
  3. 07 Apr, 2022 1 commit
  5. 04 Apr, 2022 1 commit
  7. 01 Apr, 2022 1 commit
  9. 22 Mar, 2022 1 commit
  11. 15 Mar, 2022 1 commit
  13. 04 Mar, 2022 1 commit
  15. 15 Feb, 2022 1 commit
  17. 09 Feb, 2022 1 commit
  19. 08 Feb, 2022 2 commits
  21. 04 Feb, 2022 1 commit
  23. 31 Jan, 2022 1 commit
  25. 28 Jan, 2022 1 commit
  27. 26 Jan, 2022 1 commit
  29. 20 Jan, 2022 3 commits
    • Philip Meier's avatar
      allow single extension as str in make_dataset (#5229) · 6512146e
      Philip Meier authored 
      * allow single extension as str in make_dataset

* remove test class

* remove regex

* revert collection to tuple

* cleanup
      6512146e
    • Nicolas Hug's avatar
      Cleanups for FLAVA datasets (#5164) · e047623a
      Nicolas Hug authored 
      * Change default of download for Food101 and DTD

* Set download default to False and put it at the end

* Keep stuff private

* GTSRB: train -> split. Also use pathlib

* mypy

* Remove split and partition for SUN397

* mypy

* mypy

* move download param for SST2

* Use make_dataset in SST2

* Use a base URL for GTSRB

* Let's make this code more complictaed than it needs to be because why not
      e047623a
    • Joao Gomes's avatar
      Add support for Rendered sst2 dataset (#5220) · e32b19e1
      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 RenderedSST2 dataset

* Address PR comments

* Fix bug in dataset verification
Co-authored-by: default avatarNicolas Hug <contact@nicolas-hug.com>
      e32b19e1
  31. 18 Jan, 2022 2 commits
    • Abhijit Deo's avatar
      Add support for Stanford cars dataset (#5166) · 8ba482a6
      Abhijit Deo authored 
      

* [WIP]
*added stanford_cars

* [WIP]
added stanfordCars to docs

* [WIP]
minor edits

* [WIP]
minor edits

* edited StanfordCars class

* Adding Testcase for stanford cars

* Added Testcase for stanford cars

* Added Testcase for stanford cars

* minor edit

* made changes as per the suggestions

* fixed typo in naming stanford_cars.py

* cars_meta.mat file will be created in test

* Some cleanups

* Sigh

* don't convert to strings
Co-authored-by: default avatarNicolas Hug <nicolashug@fb.com>
      8ba482a6
    • F-G Fernandez's avatar
      Adds support for EuroSAT dataset (#5114) · 57a77c45
      F-G Fernandez authored 
      

* feat: Added EuroSAT dataset

* test: Added unittest

* docs: Improved comments

* docs: Updated the documentation

* docs: Removed unnecessary comments

* fix: Fixed class implementation

* test: Fixed unittest

* fix: Fixed magic method len

* test: Fixed unittest

* refactor: Refactored EuroSAT

* refactor: Applied modifications

* Apply suggestions from code review

* refactor: Applied request changes

* refactor: Made var explicit

* fix: Fixed attribute initialization order

* refactor: Removed name mapping
Co-authored-by: default avatarPhilip Meier <github.pmeier@posteo.de>
      57a77c45
  33. 17 Jan, 2022 1 commit
  35. 14 Jan, 2022 1 commit
  37. 12 Jan, 2022 1 commit
    • Zhiqiang Wang's avatar
      Add Flowers102 dataset (#5177) · a8f2dedb
      Zhiqiang Wang authored 
      

* Add Flowers102 datasets

* Fix initialization of images and labels

* Fix _check_exists in Flowers102

* Add Flowers102 to datasets and docs

* Add Flowers102TestCase to unittest

* Fixing Python type statically

* Shuffle the fake labels

* Update test/test_datasets.py
Co-authored-by: default avatarPhilip Meier <github.pmeier@posteo.de>

* Apply the suggestions by pmeier

* Use check_integrity to check file existence

* Save the labels to base_folder

* Minor fixes

* Using a loop makes this more concise without reducing readability
Co-authored-by: default avatarPhilip Meier <github.pmeier@posteo.de>

* Using a loop makes this more concise without reducing readability
Co-authored-by: default avatarPhilip Meier <github.pmeier@posteo.de>

* Remove self.labels and self.label_to_index attributes

* minor simplification

* Check the exitence of image folder

* Revert the check

* Check the existence of image folder

* valid -> val

* keep some stuff private

* minor doc arrangements

* remove default FEATURE_TYPES

* Simplify the datasets existence

* check if the image folder exists
Co-authored-by: default avatarPhilip Meier <github.pmeier@posteo.de>

* isdir -> is_dir
Co-authored-by: default avatarPhilip Meier <github.pmeier@posteo.de>
Co-authored-by: default avatarNicolas Hug <nicolashug@fb.com>
      a8f2dedb
  39. 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
  41. 07 Jan, 2022 1 commit
  43. 06 Jan, 2022 4 commits
  45. 05 Jan, 2022 2 commits
  47. 27 Dec, 2021 1 commit
  49. 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
  51. 16 Dec, 2021 1 commit
  53. 09 Dec, 2021 1 commit
  55. 08 Dec, 2021 3 commits
  57. 30 Nov, 2021 1 commit
  59. 23 Nov, 2021 1 commit