1. 12 Jul, 2019 3 commits
  2. 11 Jul, 2019 1 commit
  3. 10 Jul, 2019 1 commit
    • ekka's avatar
      Add checks to roi_heads in detection module (#1091) · 6693b2c6
      ekka authored
      * add float32 to keypoint_rcnn docs
      
      * add float32 to faster_rcnn docs
      
      * add float32 to mask_rcnn
      
      * Update faster_rcnn.py
      
      * Update keypoint_rcnn.py
      
      * Update mask_rcnn.py
      
      * Update faster_rcnn.py
      
      * make keypoints float
      
      * make masks uint8
      
      * Update keypoint_rcnn.py
      
      * make labels Int64
      
      * make labels Int64
      
      * make labels Int64
      
      * Add checks for boxes, labels, masks, keypoints
      
      * update mask dim
      
      * remove dtype
      
      * check only if targets is not None
      
      * account for targets being a list
      
      * update target to be list of dict
      
      * Update faster_rcnn.py
      
      * Update keypoint_rcnn.py
      
      * allow boxes to be of float16 type as well
      
      * remove checks on mask
      6693b2c6
  4. 09 Jul, 2019 1 commit
  5. 08 Jul, 2019 1 commit
  6. 06 Jul, 2019 1 commit
    • Zhun Zhong's avatar
      Fix bug to RandomErasing (#1095) · 34833427
      Zhun Zhong authored
      * Fix bug to Random Erasing
      
      1. Avoid forever loop for getting parameters of erase.
      2. replace' img_b' by 'img_c', because it indicates the channel.
      3. replace v = torch.rand([img_c, h, w]) by v = torch.empty([img_c, h, w], dtype=torch.float32).normal_(). Normally distributed achieves better performance.
      
      * add test
      
      * Update test_transforms.py
      
      * Update transforms.py
      
      * Update test_transforms.py
      
      * Update transforms.py
      
      * Update functional.py
      34833427
  7. 05 Jul, 2019 2 commits
  8. 04 Jul, 2019 5 commits
  9. 03 Jul, 2019 1 commit
  10. 02 Jul, 2019 2 commits
    • yaysummeriscoming's avatar
      Fixed width multiplier (#1005) · 8350645b
      yaysummeriscoming authored
      * Fixed width multiplier
      
      Layer channels are now rounded to a multiple of 8, as per the official tensorflow implementation.  I found this fix when looking through: https://github.com/d-li14/mobilenetv2.pytorch
      
      * Channel multiple now a user configurable option
      
      The official tensorflow slim mobilenet v2 implementation rounds the number of channels in each layer to a multiple of 8.  This is now user configurable - 1 turns off rounding
      
      * Fixed whitespace error
      
      Fixed error: ./torchvision/models/mobilenet.py:152:1: W293 blank line contains whitespace
      8350645b
    • Francisco Massa's avatar
      Adds video reading / saving functionalities (#1039) · d293c4c5
      Francisco Massa authored
      * WIP
      
      * WIP
      
      * Add some documentation
      
      * Improve tests and add GC collection
      
      * [WIP] add timestamp getter
      
      * Bugfixes
      
      * Improvements and travis
      
      * Add audio fine-grained alignment
      
      * More doc
      
      * Remove unecessary file
      
      * Remove comment
      
      * Lazy import av
      
      * Remove hard-coded constants for the test
      
      * Return info stats from read
      
      * Fix for Python-2
      d293c4c5
  11. 01 Jul, 2019 1 commit
  12. 29 Jun, 2019 1 commit
  13. 28 Jun, 2019 2 commits
  14. 26 Jun, 2019 3 commits
  15. 25 Jun, 2019 1 commit
  16. 24 Jun, 2019 5 commits
    • Francisco Massa's avatar
      8616227c
    • Francisco Massa's avatar
      03e25734
    • 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
    • Michael Kösel's avatar
      d38600fe
    • 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
  17. 20 Jun, 2019 3 commits
  18. 19 Jun, 2019 1 commit
    • philipNoonan's avatar
      Enabling exporting symbols on windows (#1035) · f6262182
      philipNoonan authored
      * Enabling exporting symbols on windows
      
      Small fix to allow for the built library to be used in windows #728
      
      * added macro to allow for exported symbols on windows
      
      * added macro to allow for exported symbols on windows
      
      * removed cmake command
      
      * added dllimport using torchvision_EXPORTS preprocessor
      f6262182
  19. 18 Jun, 2019 1 commit
  20. 17 Jun, 2019 1 commit
  21. 14 Jun, 2019 3 commits