- 04 Apr, 2019 2 commits
-
-
Philip Meier authored
* introduced new super class for all vision datasets * Removed root from repr if dataset has no root * reverted some overly-ambitious autoformatting * reverted some overly-ambitious autoformatting * added split attribute to repr of STL10 dataset * made Python2 friendly and more robust * Fixed call of the superclass constructor * moved transform and target_transform back to the base classes * added check if transforms are present before printing to avoid setting them within the constructor * added missing transforms and target_transforms to base classes * fixed linter error
-
Sepehr Sameni authored
* add aux_logits support to inception it is related to pytorch/pytorch#18668 * instantiate InceptionAux only when requested it is related to pytorch/pytorch#18668 * revert googlenet * support and aux_logits in pretrained models * return namedtuple when aux_logit is True
-
- 03 Apr, 2019 1 commit
-
-
James Hamm authored
str.endswith can take a single string, or a tuple of strings. This replaces a loop over each extension and a call to endswith with a single call to endswith passing in all the allowed extensions. This has the advantage that the loop over each extension is done in c rather than python, and the code is a little less verbose.
-
- 02 Apr, 2019 2 commits
-
-
Francisco Massa authored
* Add groups support to ResNet * Kill BaseResNet * Make it support multi-machine training
-
Surgan Jandial authored
Making references/classification/train.py and references/classification/utils.py compatible with python2 (#831) * linter fixes * linter fixes
-
- 01 Apr, 2019 1 commit
-
-
Sepehr Sameni authored
* remove duplicate code from densenet * correct indentation
-
- 29 Mar, 2019 3 commits
-
-
Michael Kösel authored
-
Michael Kösel authored
* Match Tensorflows implementation of GoogLeNet * just disable the branch when pretrained is true * don't use legacy code
-
Parth Agarwal authored
Changed 'channely' to 'channel' in the comments of normalize function.
-
- 28 Mar, 2019 1 commit
-
-
Francisco Massa authored
* Add MobileNet V2 * Remove redundant functions and make tests pass * Simplify a bit the implementation * Reuse ConvBNReLU more often * Remove input_size and minor changes * Py2 fix
-
- 26 Mar, 2019 5 commits
-
-
ekka authored
-
ekka authored
-
Francisco Massa authored
-
vfdev authored
* Added Semantic Boundaries Dataset * Updated docs and datasets importing module * Improved docs
-
ekka authored
-
- 25 Mar, 2019 1 commit
-
-
ekka authored
* Add Affinetransformation Add Affinetransformation to superseed LinearTransformation * Add test * Add zero mean_vector in LinearTransformation and improved docs * update * minor fix * minor fix2 * fixed flake8 * fix flake8 * fixed transpose syntax * fixed shape of mean_vector in test * fixed test * print est cov and mean * fixed flake8 * debug * reduce num_samples * debug * fixed num_features * fixed rtol for cov * fix __repr__ * Update transforms.py * Update test_transforms.py * Update transforms.py * fix flake8 * Update transforms.py * Update transforms.py * Update transforms.py * Update transforms.py * Changed dim of mean_vector to 1D, doc and removed .numpy () from format_string * Restore test_linear_transformation() * Update test_transforms.py
-
- 24 Mar, 2019 1 commit
-
-
ekka authored
-
- 20 Mar, 2019 1 commit
-
-
Michał Zientkiewicz authored
Signed-off-by:Michal Zientkiewicz <michalz@nvidia.com>
-
- 19 Mar, 2019 1 commit
-
-
Philip Meier authored
* initial commit * fixed Python2 issue * fixed naming incorrectness and Python2 compability * fixed preparation of train folder * removed detection dataset * added docstring and repr * moved import of scipy to make the import of torchvision independent of it * improved conversion from class string to index * removed support for other years than 2012 * removed accidentally added file * moved emptying of split folder to avoid accidental deletion * removed deletion of the images * removed error conversion for Python2 * Aligned class indices with the indices identified by ImageFolder class
-
- 11 Mar, 2019 3 commits
-
-
ekka authored
In reference to #729 added comments to clarify the naming and action of the layers performing downsampling in resnets.
-
ekka authored
-
ekka authored
* Update transform doc with random offset of padding due to pad_if_needed Updating transform docs with the random offset of padding when pad_if_needed is set to True. * Update docs
-
- 09 Mar, 2019 4 commits
-
-
Will Frey authored
Raise an `IndexError` for `FakeData.__getitem__()` if the index would be out of range.
-
ekka authored
* Added dimensions in the comments The update provides the dimensions of the processed data following the style of inceptionV3 implementation. * Changed docs and comments Updated doc with the argument `transform_input`. Modified comments to match inceptionV3 style.
-
ekka authored
Including the `transform_input` argument in the docs of inceptionV3
-
Irvin Ho authored
-
- 07 Mar, 2019 3 commits
-
-
Bryan He authored
* Add Caltech101 and Caltech256 * Add information about default for target_type * Fix docs * Add function to download from Google Drive * Add CelebA dataset * Only import pandas when needed * Addressing comments * Remove trailing whitespace * Replace torch.LongTensor with torch.as_tensor
-
maria8899 authored
-
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
-
- 01 Mar, 2019 1 commit
-
-
Francisco Massa authored
* Fixes for PyTorch version of tqdm * Flake * Flake fix
-
- 28 Feb, 2019 2 commits
-
-
Francisco Massa authored
-
Surgan Jandial authored
-
- 27 Feb, 2019 1 commit
-
-
Soumith Chintala authored
-
- 26 Feb, 2019 1 commit
-
-
Surgan Jandial authored
-
- 21 Feb, 2019 1 commit
-
-
Tongzhou Wang authored
-
- 18 Feb, 2019 2 commits
-
-
surgan12 authored
* randomresizedmods * lint checks * test to randomrescrop added * updates * tests updated * tests updated * upd * updates * Update torchvision/transforms/transforms.py Co-Authored-By:surgan12 <33121121+surgan12@users.noreply.github.com> * tests changed * trvis * travis * fixes syntax * ... * flake fixes * flake_fixes * flake_fixes2
-
surgan12 authored
* flake_fixes * flake_fixes2
-
- 14 Feb, 2019 3 commits
-
-
ekka authored
* Modifying the comments of inceptionV3 dimensions Modifying the comments of inceptionV3 dimensions to match the pytorch convention. Relevant (https://github.com/pytorch/vision/pull/719#pullrequestreview-203194302) * Added Batch size in comment * Update inception.py
-
ekka authored
* Updated inceptionV3 to accept different sized images (Adaptive avg pool) The update allows inceptionV3 to process images larger or smaller than prescribed image size (299x299) using adaptive average pooling. Will be useful while finetuning or testing on different resolution images. * Update inception.py
-
ekka authored
The update allows VGG to process images larger or smaller than prescribed imagenet size using adaptive average pooling. Will be useful while finetuning or testing on different resolution images. Similar to https://github.com/pytorch/vision/pull/643 and https://github.com/pytorch/vision/pull/672. I did not include adaptive avg pool in features or classifier block so that these predefined blocks can be used as it is.
-