1. 08 Feb, 2022 3 commits
  2. 03 Feb, 2022 1 commit
  3. 02 Feb, 2022 1 commit
  4. 01 Feb, 2022 2 commits
  5. 27 Jan, 2022 1 commit
  6. 23 Jan, 2022 1 commit
  7. 21 Jan, 2022 1 commit
    • Hu Ye's avatar
      add FCOS (#4961) · 7d4bdd43
      Hu Ye authored
      
      
      * add fcos
      
      * update fcos
      
      * add giou_loss
      
      * add BoxLinearCoder for FCOS
      
      * add full code for FCOS
      
      * add giou loss
      
      * add fcos
      
      * add __all__
      
      * Fixing lint
      
      * Fixing lint in giou_loss.py
      
      * Add typing annotation to fcos
      
      * Add trained checkpoints
      
      * Use partial to replace lambda
      
      * Minor fixes to docstrings
      
      * Apply ufmt format
      
      * Fixing docstrings
      
      * Fixing jit scripting
      
      * Minor fixes to docstrings
      
      * Fixing jit scripting
      
      * Ignore mypy in fcos
      
      * Fixing trained checkpoints
      
      * Fixing unit-test of jit script
      
      * Fixing docstrings
      
      * Add test/expect/ModelTester.test_fcos_resnet50_fpn_expect.pkl
      
      * Fixing test_detection_model_trainable_backbone_layers
      
      * Update test_fcos_resnet50_fpn_expect.pkl
      
      * rename stride to box size
      
      * remove TODO and fix some typo
      
      * merge some code for better
      
      * impove the comments
      
      * remove decode and encode of BoxLinearCoder
      
      * remove some unnecessary hints
      
      * use the default value in detectron2.
      
      * update doc
      
      * Add unittest for BoxLinearCoder
      
      * Add types in FCOS
      
      * Add docstring for BoxLinearCoder
      
      * Minor fix for the docstring
      
      * update doc
      
      * Update fcos_resnet50_fpn_coco pretained weights url
      
      * Update torchvision/models/detection/fcos.py
      Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
      
      * Update torchvision/models/detection/fcos.py
      Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
      
      * Update torchvision/models/detection/fcos.py
      Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
      
      * Update torchvision/models/detection/fcos.py
      Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
      
      * Add FCOS model documentation
      
      * Fix typo in FCOS documentation
      
      * Add fcos to the prototype builder
      
      * Capitalize COCO_V1
      
      * Fix params of fcos
      
      * fix bug for partial
      
      * Fixing docs indentation
      
      * Fixing docs format in giou_loss
      
      * Adopt Reference for GIoU Loss
      
      * Rename giou_loss to generalized_box_iou_loss
      
      * remove overwrite_eps
      
      * Update AP test values
      
      * Minor fixes for the docs
      
      * Minor fixes for the docs
      
      * Update torchvision/models/detection/fcos.py
      Co-authored-by: default avatarZhiqiang Wang <zhiqwang@foxmail.com>
      
      * Update torchvision/prototype/models/detection/fcos.py
      Co-authored-by: default avatarZhiqiang Wang <zhiqwang@foxmail.com>
      Co-authored-by: default avatarzhiqiang <zhiqwang@foxmail.com>
      Co-authored-by: default avatarJoao Gomes <jdsgomes@fb.com>
      Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
      Co-authored-by: default avatarJoao Gomes <joaopsgomes@gmail.com>
      7d4bdd43
  8. 20 Jan, 2022 2 commits
    • Vasilis Vryniotis's avatar
      Adding ConvNeXt architecture in prototype (#5197) · afda28ac
      Vasilis Vryniotis authored
      * Adding CNBlock and skeleton architecture
      
      * Completed implementation.
      
      * Adding model in prototypes.
      
      * Add test and minor refactor for JIT.
      
      * Fix mypy.
      
      * Fixing naming conventions.
      
      * Fixing tests.
      
      * Fix stochastic depth percentages.
      
      * Adding stochastic depth to tiny variant.
      
      * Minor refactoring and adding comments.
      
      * Adding weights.
      
      * Update default weights.
      
      * Fix transforms issue
      
      * Move convnext to prototype.
      
      * linter fix
      
      * fix docs
      
      * Addressing code review comments.
      afda28ac
    • 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
  9. 19 Jan, 2022 1 commit
  10. 18 Jan, 2022 3 commits
    • Nicolas Hug's avatar
      Change order of sections in toctree (#5195) · bf073e78
      Nicolas Hug authored
      bf073e78
    • 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
  11. 17 Jan, 2022 1 commit
  12. 14 Jan, 2022 1 commit
  13. 13 Jan, 2022 1 commit
  14. 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
  15. 10 Jan, 2022 2 commits
    • Yiwen Song's avatar
      [ViT] Graduate ViT from prototype (#5173) · 68f511eb
      Yiwen Song authored
      * graduate vit from prototype
      
      * nit
      
      * add vit to docs and hubconf
      
      * ufmt
      
      * re-correct ufmt
      
      * again
      
      * fix linter
      68f511eb
    • 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
  16. 07 Jan, 2022 1 commit
  17. 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
  18. 05 Jan, 2022 2 commits
  19. 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
  20. 06 Dec, 2021 1 commit
  21. 12 Nov, 2021 2 commits
  22. 09 Nov, 2021 1 commit
    • Aditya Oke's avatar
      Add utility to draw keypoints (#4216) · a4ca717f
      Aditya Oke authored
      * fix
      
      * Outline Keypoints API
      
      * Add utility
      
      * make it work :)
      
      * Fix optional type
      
      * Add connectivity, fmassa's advice 😃
      
      
      
      * Minor code improvement
      
      * small fix
      
      * fix implementation
      
      * Add tests
      
      * Fix tests
      
      * Update colors
      
      * Fix bug and test more robustly
      
      * Add a comment, merge stuff
      
      * Fix fmt
      
      * Support single str for merging
      
      * Remove unnecessary vars.
      Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
      a4ca717f
  23. 05 Nov, 2021 2 commits
  24. 04 Nov, 2021 2 commits
  25. 03 Nov, 2021 1 commit
  26. 02 Nov, 2021 2 commits
  27. 01 Nov, 2021 1 commit