1. 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
  2. 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
  3. 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
  4. 05 Jun, 2019 1 commit
  5. 03 Jun, 2019 1 commit
  6. 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
  7. 22 May, 2019 1 commit
  8. 21 May, 2019 3 commits
  9. 19 May, 2019 1 commit
  10. 01 May, 2019 1 commit
  11. 30 Apr, 2019 1 commit
    • Bar's avatar
      Add ShuffleNet v2 (#849) · 7a4845a9
      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
      7a4845a9
  12. 18 Apr, 2019 1 commit
  13. 26 Mar, 2019 1 commit
  14. 07 Mar, 2019 1 commit
    • Michael Kösel's avatar
      Add GoogLeNet (Inception v1) (#678) · a2093007
      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
      a2093007
  15. 23 Feb, 2019 1 commit
  16. 18 Feb, 2019 1 commit
  17. 25 Dec, 2018 1 commit
  18. 21 Dec, 2018 1 commit
  19. 13 Dec, 2018 1 commit
    • surgan12's avatar
      docs changed (#685) · 0ce81569
      surgan12 authored
      * pad updated
      
      * checked pad
      
      * pad checked
      
      * pad checked
      
      * docs changed
      0ce81569
  20. 06 Dec, 2018 1 commit
    • Benjamin Pinaya's avatar
      VOCSegmentation, VOCDetection, linting passing, examples. (#663) · e489abc6
      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: default avatarbpinaya <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.
      e489abc6
  21. 04 Dec, 2018 2 commits
  22. 11 Oct, 2018 1 commit
  23. 13 Sep, 2018 1 commit
  24. 15 May, 2018 1 commit
  25. 07 Apr, 2018 1 commit
  26. 27 Mar, 2018 1 commit
  27. 10 Mar, 2018 1 commit
    • Frédérik Paradis's avatar
      Adding a DatasetFolder class. (#444) · fa377c47
      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
      fa377c47
  28. 09 Mar, 2018 2 commits
  29. 08 Mar, 2018 1 commit
    • Francisco Massa's avatar
      Update link of ImageNet preprocessing · 456d3b97
      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).
      456d3b97
  30. 06 Feb, 2018 1 commit
  31. 07 Dec, 2017 1 commit
  32. 06 Dec, 2017 1 commit
  33. 17 Nov, 2017 1 commit
  34. 07 Nov, 2017 1 commit