- 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
-
- 30 Apr, 2019 1 commit
-
-
Bar authored
* Add ShuffleNet v2 Added 4 configurations: x0.5, x1, x1.5, x2 Add 2 pretrained models: x0.5, x1 * fix lint * Change globalpool to torch.mean() call
-
- 18 Apr, 2019 1 commit
-
-
Philip Meier authored
-
- 26 Mar, 2019 1 commit
-
-
vfdev authored
* Added Semantic Boundaries Dataset * Updated docs and datasets importing module * Improved docs
-
- 07 Mar, 2019 1 commit
-
-
Michael Kösel authored
* Add GoogLeNet (Inception v1) * Fix missing padding * Add missing ReLu to aux classifier * Add Batch normalized version of GoogLeNet * Use ceil_mode instead of padding and initialize weights using "xavier" * Match BVLC GoogLeNet zero initialization of classifier * Small cleanup * use adaptive avg pool * adjust network to match TensorFlow * Update url of pre-trained model and add classification results on ImageNet * Bugfix that improves performance by 1 point
-
- 23 Feb, 2019 1 commit
-
-
Vishwak Srinivasan authored
-
- 18 Feb, 2019 1 commit
-
-
surgan12 authored
* flake_fixes * flake_fixes2
-
- 25 Dec, 2018 1 commit
-
-
hysts authored
* Add KMNIST * Add documentation * Add KMNIST to module's __all__ list
-
- 21 Dec, 2018 1 commit
-
-
Michael Kösel authored
* Add Cityscapes Dataset * Rename 'label' to 'semantic' to make the meaning more clear * Add support for gtCoarse target set
-
- 13 Dec, 2018 1 commit
-
-
surgan12 authored
* pad updated * checked pad * pad checked * pad checked * docs changed
-
- 06 Dec, 2018 1 commit
-
-
Benjamin Pinaya authored
* VOC Dataset, linted, flak8 passing, samples on gist. * Double backtick on values. * Apply suggestions from code review Add suggestions from @ellisbrown, using dict of dicts instead of array index. Co-Authored-By:bpinaya <bpg_92@hotmail.com> * Fixed errors with the new comments. * Added documentation on RST * Removed getBB, added parse_voc_xml, variable naming change. * Removed unused variable, removed VOC_CLASSES, two new gists for test.
-
- 04 Dec, 2018 2 commits
-
-
Adam J. Stewart authored
* Add Flickr8k and Flickr30k Datasets * Add Flickr to the dataset docs * Sort ids, glob during construction * annFile -> ann_file * Fix undefined variable name bug
-
Adam J. Stewart authored
* Add SBU Captioned Photo Dataset * Add SBU to the dataset docs
-
- 11 Oct, 2018 1 commit
-
-
Samuel Fadel authored
-
- 13 Sep, 2018 1 commit
-
-
Stefan Otte authored
Give a short example of how to build a transform pipeline for a segmentation task.
-
- 15 May, 2018 1 commit
-
-
Vishwak Srinivasan authored
-
- 07 Apr, 2018 1 commit
-
-
Vishwak Srinivasan authored
* Add LinearTransformation to docs - Arrange the transforms alphabetically * Add missing transforms to docs
-
- 27 Mar, 2018 1 commit
-
-
vfdev authored
Add testing case with Pillow-SIMD, fix bug in RandomAffine if Pillow<5.0.0, add missing docs for affine (#439)
-
- 10 Mar, 2018 1 commit
-
-
Frédérik Paradis authored
* Adding tests to ImageFolder * Adding DatasetFolder class * Fix tests for pytest and code for lint checker * Adding mock to requirements for ImageFolder tests * Remove mocks from requirements
-
- 09 Mar, 2018 2 commits
-
-
Soumith Chintala authored
This reverts commit ab03dc43.
-
Frédérik Paradis authored
* Adding tests to ImageFolder * Adding DatasetFolder class * Fix tests for pytest and code for lint checker * Adding mock to requirements for ImageFolder tests
-
- 08 Mar, 2018 1 commit
-
-
Francisco Massa authored
It was pointing to an old version where the validation pre-processing was the same as the training pre-processing, and thus had randomness in it (instead of CenterCrop).
-
- 06 Feb, 2018 1 commit
-
-
Kai Arulkumaran authored
-
- 07 Dec, 2017 1 commit
-
-
David de la Iglesia Castro authored
-
- 06 Dec, 2017 1 commit
-
-
Martin Raison authored
* EMNIST dataset + speedup *MNIST preprocessing
-
- 17 Nov, 2017 1 commit
-
-
Alykhan Tejani authored
* add docs for grayscale + random grayscale transforms
-
- 07 Nov, 2017 1 commit
-
-
Dr. Kashif Rasul authored
-