1. 11 May, 2020 1 commit
    • Philip Meier's avatar
      Add static type check with mypy (#2195) · a81d99b7
      Philip Meier authored
      * add mypy config
      
      * fix syntax error
      
      * fix annotations in torchvision/utils.py
      
      * add mypy type check to CircleCI
      
      * add mypy cache to ignore files
      
      * try fix CI
      
      * ignore flake8 F821 since it interferes with mypy
      
      * add mypy type check to config generator
      
      * explicitly set config files
      a81d99b7
  2. 07 May, 2020 3 commits
    • Francisco Massa's avatar
      Fix mypy type annotations (#1696) · f71316fa
      Francisco Massa authored
      
      
      * Fix mypy type annotations
      
      * follow torchscript Tuple type
      
      * redefine torch_choice output type
      
      * change the type in cached_grid_anchors
      
      * minor bug
      Co-authored-by: default avatarGuanheng Zhang <zhangguanheng@devfair0197.h2.fair>
      Co-authored-by: default avatarGuanheng Zhang <zhangguanheng@learnfair0341.h2.fair>
      f71316fa
    • Negin Raoof's avatar
      [ONNX] Fix model export for images w/ no detection (#2126) · 3ac864dc
      Negin Raoof authored
      * Fixing nms on boxes when no detection
      
      * test
      
      * Fix for scale_factor computation
      
      * remove newline
      
      * Fix for mask_rcnn dynanmic axes
      
      * Clean up
      
      * Update transform.py
      
      * Fix for torchscript
      
      * Fix scripting errors
      
      * Fix annotation
      
      * Fix lint
      
      * Fix annotation
      
      * Fix for interpolate scripting
      
      * Fix for scripting
      
      * refactoring
      
      * refactor the code
      
      * Fix annotation
      
      * Fixed annotations
      
      * Added test for resize
      
      * lint
      
      * format
      
      * bump ORT
      
      * ort-nightly version
      
      * Going to ort 1.1.0
      
      * remove version
      
      * install typing-extension
      
      * Export model for images with no detection
      
      * Upgrade ort nightly
      
      * update ORT
      
      * Update test_onnx.py
      
      * updated tests
      
      * Updated tests
      
      * merge
      
      * Update transforms.py
      
      * Update cityscapes.py
      
      * Update celeba.py
      
      * Update caltech.py
      
      * Update pkg_helpers.bash
      
      * Clean up
      
      * Clean up for dynamic split
      
      * Remove extra casts
      
      * flake8
      3ac864dc
    • Guillem Orellana Trullols's avatar
      Update ucf101.py (#2186) · 14af9de6
      Guillem Orellana Trullols authored
      Now the dataset is not working properly because of this line of code `indices = [i for i in range(len(video_list)) if video_list[i][len(self.root) + 1:] in selected_files]`. 
      Performing the `len(self.root) + 1` only make sense if there is no training / to root
      
      ```
      >>> root = 'data/ucf-101/videos'
      >>> video_path = 'data/ucf-101/videos/activity/video.avi'
      >>> video_path [len(root ):]
      '/activity/video.avi'
      >>> video_path [len(root ) + 1:]
      'activity/video.avi'
      ```
      
      Appending the root path also to the selected files is a simple solution and make the dataset works with and without a trailing slash.
      14af9de6
  3. 05 May, 2020 6 commits
  4. 04 May, 2020 5 commits
  5. 29 Apr, 2020 3 commits
  6. 28 Apr, 2020 2 commits
  7. 27 Apr, 2020 2 commits
  8. 26 Apr, 2020 1 commit
  9. 24 Apr, 2020 2 commits
  10. 23 Apr, 2020 1 commit
  11. 21 Apr, 2020 1 commit
    • Negin Raoof's avatar
      [ONNX] Fix for dynamic scale_factor export (#2087) · d6ee8757
      Negin Raoof authored
      * Fixing nms on boxes when no detection
      
      * test
      
      * Fix for scale_factor computation
      
      * remove newline
      
      * Fix for mask_rcnn dynanmic axes
      
      * Clean up
      
      * Update transform.py
      
      * Fix for torchscript
      
      * Fix scripting errors
      
      * Fix annotation
      
      * Fix lint
      
      * Fix annotation
      
      * Fix for interpolate scripting
      
      * Fix for scripting
      
      * refactoring
      
      * refactor the code
      
      * Fix annotation
      
      * Fixed annotations
      
      * Added test for resize
      
      * lint
      
      * format
      
      * bump ORT
      
      * ort-nightly version
      
      * Going to ort 1.1.0
      
      * remove version
      
      * install typing-extension
      d6ee8757
  12. 16 Apr, 2020 1 commit
  13. 15 Apr, 2020 1 commit
  14. 14 Apr, 2020 4 commits
  15. 10 Apr, 2020 2 commits
  16. 09 Apr, 2020 2 commits
  17. 08 Apr, 2020 2 commits
  18. 07 Apr, 2020 1 commit
    • Brian Hart's avatar
      improve consistency among box IoU calculations (#2072) · f6a3e0c3
      Brian Hart authored
      Torchvision includes at least 3 bits of code that calculate
      box Intersection over Union values (and usually compare to
      a threshold):
      
      - box_iou in torchvision/ops/boxes.py
      - devIoU in torchvision/csrc/cuda/nms_cuda.cu
      - nms_cpu_kernel in torchvision/csrc/cpu/nms_cpu.cpp
      
      The calculations were performed slightly differently between
      those, leading to occasional differences in results.
      
      Update devIoU to use the same method as the others for better
      consistency.
      
      This change improves agreement between the CPU and CUDA
      calculations but the results can still differ slightly.
      Setting NVCC_FLAGS to include "--fmad=true" would provide
      still better agreement, but with likely cost to performance.
      f6a3e0c3