1. 07 Oct, 2020 1 commit
    • Tejan Karmali's avatar
      Added GaussianBlur transform (#2658) · 4106dbb8
      Tejan Karmali authored
      
      
      * Added GaussianBlur transform
      
      * fixed linting
      
      * supports fixed radius for kernel
      
      * [WIP] New API for gaussian_blur
      
      * Gaussian blur with kernelsize and sigma API
      
      * Fixed implementation and updated tests
      
      * Added large input case and refactored gt into a file
      
      * Updated docs
      
      * fix kernel dimesnions order while creating kernel
      
      * added tests for exception handling of gaussian blur
      
      * fix linting, bug in tests
      
      * Fixed failing tests, refactored code and other minor fixes
      Co-authored-by: default avatarvfdev-5 <vfdev.5@gmail.com>
      4106dbb8
  2. 06 Oct, 2020 1 commit
  3. 05 Oct, 2020 1 commit
  4. 01 Oct, 2020 1 commit
    • Aditya Oke's avatar
      Adds bounding boxes conversion (#2710) · e70c91a9
      Aditya Oke authored
      
      
      * adds boxes conversion
      
      * adds documentation
      
      * adds xywh tests
      
      * fixes small typo
      
      * adds tests
      
      * Remove sphinx theme
      
      * corrects assertions
      
      * cleans code as per suggestion
      Signed-off-by: default avatarAditya Oke <okeaditya315@gmail.com>
      
      * reverts assertion
      
      * fixes to assertEqual
      
      * fixes inplace operations
      
      * Adds docstrings
      
      * added documentation
      
      * changes tests
      
      * moves code to box_convert
      
      * adds more tests
      
      * Apply suggestions from code review
      
      Let's leave those changes to a separate PR
      
      * fixes documentation
      Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
      e70c91a9
  5. 24 Sep, 2020 1 commit
    • Aditya Oke's avatar
      Adds Generalized IOU (#2642) · 6e10e3f8
      Aditya Oke authored
      * tries adding genaralized_iou
      
      * fixes linting
      
      * Adds docs for giou, iou and box area
      
      * fixes lint
      
      * removes docs to fixup in other PR
      
      * linter fix
      
      * Cleans comments
      
      * Adds tests for box area, iou and giou
      
      * typo fix for testCase
      
      * fixes typo
      
      * fixes box area test
      
      * fixes implementation
      
      * updates tests to tolerance
      6e10e3f8
  6. 23 Sep, 2020 1 commit
    • vfdev's avatar
      Normalize, LinearTransformation are scriptable (#2645) · 1b415254
      vfdev authored
      * [WIP] All transforms are now derived from torch.nn.Module
      - Compose, RandomApply, Normalize can be jit scripted
      
      * Fixed flake8
      
      * Updated code and docs
      - added getattr to Lambda and tests
      - updated code and docs of Compose
      - added failing test with append/extend on Composed.transforms
      
      * Fixed flake8
      
      * Updated code, tests and docs
      1b415254
  7. 18 Sep, 2020 2 commits
  8. 09 Sep, 2020 1 commit
  9. 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
  10. 03 Aug, 2020 1 commit
  11. 23 Jun, 2020 2 commits
  12. 15 Apr, 2020 1 commit
  13. 10 Apr, 2020 1 commit
  14. 31 Mar, 2020 2 commits
  15. 12 Mar, 2020 1 commit
  16. 25 Nov, 2019 1 commit
    • eellison's avatar
      Make maskrcnn scriptable (#1407) · d88d8961
      eellison authored
      * almost working...
      
      * respond to comments
      
      * add empty tensor op, handle different output types in generalized rcnn
      
      * clean ups
      
      * address comments
      
      * more changes
      
      * it's working!
      
      * torchscript bugs
      
      * add script/ eager test
      
      * eval script model
      
      * fix flake
      
      * division import
      
      * py2 compat
      
      * update test, fix arange bug
      
      * import division statement
      
      * fix linter
      
      * fixes
      
      * changes needed for JIT master
      
      * cleanups
      
      * remove imagelist_to
      
      * requested changes
      
      * Make FPN backwards-compatible and torchscript compatible
      
      We remove support for feature channels=0, but support for it was already a bit limited
      
      * Fix ONNX regression
      d88d8961
  17. 16 Oct, 2019 1 commit
  18. 02 Oct, 2019 1 commit
  19. 06 Aug, 2019 1 commit
  20. 02 Aug, 2019 1 commit
    • Francisco Massa's avatar
      Expose docs for io and ops package (#1189) · 4ec38d49
      Francisco Massa authored
      * Expose docs for io and ops package
      
      Had do modify the docstrings to use Napoleon NumPy style, because Napoleon Google Style doesn't support multiple return arguments
      
      * Add video section
      4ec38d49
  21. 01 Aug, 2019 1 commit
  22. 31 Jul, 2019 1 commit
  23. 26 Jun, 2019 1 commit
    • Sergey Zagoruyko's avatar
      Add pretrained Wide ResNet (#912) · 2b6da28c
      Sergey Zagoruyko authored
      * add wide resnet
      
      * add docstring for wide resnet
      
      * update WRN-50-2 model
      
      * add docs
      
      * extend WRN docstring
      
      * use pytorch storage for WRN
      
      * fix rebase
      
      * fix typo in docs
      2b6da28c
  24. 24 Jun, 2019 3 commits
    • Zhun Zhong's avatar
      transforms: add Random Erasing for image augmentation (#909) · 3254560b
      Zhun Zhong authored
      * add erase function
      
      * add Random Erasing
      
      * Update transforms.py
      
      * Update transforms.py
      
      * add test for random erasing
      
      * Update test_transforms.py
      
      * fix flake8
      
      * Update test_transforms.py
      
      * Update functional.py
      
      * Update test_transforms.py
      
      * fix bug for per-pixel erasing
      
      * Update transforms.py
      
      * specific for coordinate (x, y)
      
      * add raise TypeError for img
      
      * Update transforms.py
      
      * Update transforms.rst
      3254560b
    • Dmitry Belenko's avatar
      Implementation of the MNASNet family of models (#829) · 69b28578
      Dmitry Belenko authored
      * Add initial mnasnet impl
      
      * Remove all type hints, comply with PyTorch overall style
      
      * Expose models
      
      * Remove avgpool from features() and add separately
      
      * Fix python3-only stuff, replace subclasses with functions
      
      * fix __all__
      
      * Fix typo
      
      * Remove conditional dropout
      
      * Make dropout functional
      
      * Addressing @fmassa's feedback, round 1
      
      * Replaced adaptive avgpool with mean on H and W to prevent collapsing the batch dimension
      
      * Partially address feedback
      
      * YAPF
      
      * Removed redundant class vars
      
      * Update urls to releases
      
      * Add information to models.rst
      
      * Replace init with kaiming_normal_ in fan-out mode
      
      * Use load_state_dict_from_url
      69b28578
    • Adrian Campos's avatar
      Corrected typo: 5 to 0.5 (#1041) · 12fab3a2
      Adrian Campos authored
      Values from random.random() are always in the range [0.0, 1.0), so the statement `random.random() > 5` will never be true.
      12fab3a2
  25. 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
  26. 05 Jun, 2019 1 commit
  27. 03 Jun, 2019 1 commit
  28. 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
  29. 22 May, 2019 1 commit
  30. 21 May, 2019 3 commits
  31. 19 May, 2019 1 commit
  32. 01 May, 2019 1 commit