1. 23 Jun, 2020 2 commits
  2. 15 Apr, 2020 1 commit
  3. 10 Apr, 2020 1 commit
  4. 31 Mar, 2020 2 commits
  5. 12 Mar, 2020 1 commit
  6. 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
  7. 16 Oct, 2019 1 commit
  8. 02 Oct, 2019 1 commit
  9. 06 Aug, 2019 1 commit
  10. 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
  11. 01 Aug, 2019 1 commit
  12. 31 Jul, 2019 1 commit
  13. 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
  14. 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
  15. 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
  16. 05 Jun, 2019 1 commit
  17. 03 Jun, 2019 1 commit
  18. 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
  19. 22 May, 2019 1 commit
  20. 21 May, 2019 3 commits
  21. 19 May, 2019 1 commit
  22. 01 May, 2019 1 commit
  23. 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
  24. 18 Apr, 2019 1 commit
  25. 26 Mar, 2019 1 commit
  26. 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
  27. 23 Feb, 2019 1 commit
  28. 18 Feb, 2019 1 commit
  29. 25 Dec, 2018 1 commit
  30. 21 Dec, 2018 1 commit
  31. 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
  32. 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
  33. 04 Dec, 2018 2 commits