Skip to content

GitLab

Switch branch/tag
  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
  3. 07 Jan, 2022 1 commit
  5. 06 Jan, 2022 4 commits
  7. 05 Jan, 2022 2 commits
  9. 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
  11. 12 Nov, 2021 1 commit
  13. 09 Nov, 2021 2 commits
  15. 08 Nov, 2021 1 commit
  17. 05 Nov, 2021 2 commits
  19. 04 Nov, 2021 1 commit
  21. 28 Oct, 2021 1 commit
  23. 04 Oct, 2021 1 commit
    • Philip Meier's avatar
      Add ufmt (usort + black) as code formatter (#4384) · 5f0edb97
      Philip Meier authored 
      

* add ufmt as code formatter

* cleanup

* quote ufmt requirement

* split imports into more groups

* regenerate circleci config

* fix CI

* clarify local testing utils section

* use ufmt pre-commit hook

* split relative imports into local category

* Revert "split relative imports into local category"

This reverts commit f2e224cde2008c56c9347c1f69746d39065cdd51.

* pin black and usort dependencies

* fix local test utils detection

* fix ufmt rev

* add reference utils to local category

* fix usort config

* remove custom categories sorting

* Run pre-commit without fixing flake8

* got a double import in merge
Co-authored-by: default avatarNicolas Hug <nicolashug@fb.com>
      5f0edb97
  25. 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
  27. 08 Sep, 2021 1 commit
  29. 06 Aug, 2021 1 commit
  31. 28 Jul, 2021 1 commit
  33. 01 Jul, 2021 1 commit
  35. 10 Jun, 2021 1 commit
  37. 27 Apr, 2021 1 commit
  39. 26 Apr, 2021 2 commits
  41. 23 Apr, 2021 2 commits
  43. 22 Apr, 2021 1 commit
  45. 20 Apr, 2021 1 commit
  47. 12 Apr, 2021 2 commits
  49. 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
  51. 30 Mar, 2021 2 commits
  53. 25 Mar, 2021 1 commit
  55. 24 Mar, 2021 2 commits
    • Philip Meier's avatar
      add tests for MNIST and variants (#3423) · aa365993
      Philip Meier authored 
      * add tests for MNIST and variants

* remove old tests and fakedata generation

* fix default config detection for if dataset has variable keywords

* use split="mnist" as default for EMNIST

* fix QMNIST tests

* lint

* fix special kwargs detection

* Revert "use split="mnist" as default for EMNIST"

This reverts commit 62c9b23597f4a391cad409cbd93edac1b3474acf.

* fix tests

* fix QMNIST test case name

* remove dead code from test

* Revert "remove old tests and fakedata generation"

This reverts commit a285b97c4827566a5bc06c288f5bba8d614a4f7a.

* remove old tests

* readd removed import
      aa365993
    • Philip Meier's avatar
      Improve error handling in make_dataset (#3496) · 0818c682
      Philip Meier authored 
      * factor out find_classes

* use find_classes in video datasets

* adapt old tests
      0818c682
  57. 22 Mar, 2021 1 commit