- 07 Oct, 2020 1 commit
-
-
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:vfdev-5 <vfdev.5@gmail.com>
-
- 06 Oct, 2020 1 commit
-
-
Edgar Andrés Margffoy Tuay authored
* Expose image io functions in docs * Expose image funcs in module * Update docstring * Update docstrings for all functions * Update section header
-
- 05 Oct, 2020 1 commit
-
-
vfdev authored
- Fixed incorrect docs layout and a warning
-
- 01 Oct, 2020 1 commit
-
-
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:
Aditya 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:
Francisco Massa <fvsmassa@gmail.com>
-
- 24 Sep, 2020 1 commit
-
-
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
-
- 23 Sep, 2020 1 commit
-
-
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
-
- 18 Sep, 2020 2 commits
-
-
Aditya Oke authored
* Adds docs and fixes improts for ops and io * reverts io and init for images
-
vfdev authored
* [WIP] Added tests on batch of tensors * Updated tests on batch of images * All functional transforms can work with (..., C, H, W) format * Added transforms tests on batch tensors * Added batch tests for five/ten crop - updated docs
-
- 09 Sep, 2020 1 commit
-
-
Adam J. Stewart authored
-
- 25 Aug, 2020 2 commits
-
-
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
-
Marijan Smetko authored
-
- 03 Aug, 2020 1 commit
-
-
Zhiqing Xiao authored
Add document entry for datasets.Omniglot
-
- 23 Jun, 2020 2 commits
-
-
Zhiqiang Wang authored
* Update docs * Update docs/source/ops.rst Co-authored-by:
Francisco Massa <fvsmassa@gmail.com> Co-authored-by:
Francisco Massa <fvsmassa@gmail.com>
-
Zhiqiang Wang authored
* Add Deformable Convolution in Doc * Add PSRoIPool and PSRoIAlign in Doc
-
- 15 Apr, 2020 1 commit
-
-
Edgar Riba authored
-
- 10 Apr, 2020 1 commit
-
-
moto authored
-
- 31 Mar, 2020 2 commits
-
-
Philip Meier authored
* guess documentation location * document mean/std origins * fix artifact * update for clarity * fix typo
-
Brian Johnson authored
* Update requirements.txt * Update conf.py * Update conf.py
-
- 12 Mar, 2020 1 commit
-
-
NVS Abhilash authored
-
- 25 Nov, 2019 1 commit
-
-
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
-
- 16 Oct, 2019 1 commit
-
-
Surgan Jandial authored
-
- 02 Oct, 2019 1 commit
-
-
Stefan Otte authored
-
- 06 Aug, 2019 1 commit
-
-
Francisco Massa authored
* Add docs for video models * Fix docstrings for resnet and vgg
-
- 02 Aug, 2019 1 commit
-
-
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
-
- 01 Aug, 2019 1 commit
-
-
Francisco Massa authored
* Add docs to UCF101 and HMDB51 datasets * Remove extra line * Lint
-
- 31 Jul, 2019 1 commit
-
-
Francisco Massa authored
Also add docs
-
- 26 Jun, 2019 1 commit
-
-
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
-
- 24 Jun, 2019 3 commits
-
-
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
-
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
-
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.
-
- 12 Jun, 2019 1 commit
-
-
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
-
- 05 Jun, 2019 1 commit
-
-
hirune924 authored
-
- 03 Jun, 2019 1 commit
-
-
Brian Johnson authored
-
- 27 May, 2019 1 commit
-
-
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
-
- 22 May, 2019 1 commit
-
-
Sergey Zagoruyko authored
-
- 21 May, 2019 3 commits
-
-
Francisco Massa authored
-
Francisco Massa authored
-
Francisco Massa authored
-
- 19 May, 2019 1 commit
-
-
Francisco Massa authored
Also move weights from ShuffleNet to PyTorch bucket. Additionally, rename shufflenet to make it consistent with the other models
-
- 01 May, 2019 1 commit
-
-
Surgan Jandial authored
-