1. 13 Mar, 2020 1 commit
  2. 10 Mar, 2020 1 commit
  3. 04 Mar, 2020 1 commit
  4. 14 Feb, 2020 1 commit
  5. 13 Feb, 2020 1 commit
  6. 04 Feb, 2020 1 commit
    • F-G Fernandez's avatar
      Added __repr__ attribute to GeneralizedRCNNTransform (#1834) · e2573a71
      F-G Fernandez authored
      * feat: Added __repr__ attribute to GeneralizedRCNNTransform
      
      Added more details to default __repr__ attribute for printing.
      
      * fix: Put back relative imports
      
      * style: Fixed pep8 compliance
      
      Switched strings with  syntax to f-strings.
      
      * test: Added test for GeneralizedRCNNTransform __repr__
      
      Checked integrity of __repr__ attribute
      
      * test: Fixed unittest for __repr__
      
      Fixed the formatted strings in the __repr__ integrity check for GeneralizedRCNNTransform
      
      * fix: Fixed f-strings for earlier python versions
      
      Switched back f-strings to .format syntax for Python3.5 compatibility.
      
      * fix: Fixed multi-line string
      
      Fixed multiple-line string syntax for compatibility
      
      * fix: Fixed GeneralizedRCNNTransform unittest
      
      Fixed formatting of min_size argument of the resizing part
      e2573a71
  7. 27 Jan, 2020 1 commit
  8. 22 Jan, 2020 1 commit
  9. 17 Jan, 2020 1 commit
  10. 16 Jan, 2020 1 commit
  11. 13 Jan, 2020 1 commit
  12. 02 Jan, 2020 2 commits
  13. 17 Dec, 2019 1 commit
  14. 16 Dec, 2019 1 commit
  15. 11 Dec, 2019 1 commit
  16. 05 Dec, 2019 2 commits
  17. 30 Nov, 2019 1 commit
    • driazati's avatar
      Add tests for results in script vs eager mode (#1430) · 227027d5
      driazati authored
      * Add tests for results in script vs eager mode
      
      This copies some logic from `test_jit.py` to check that a TorchScript'ed
      model's outputs are the same as outputs from the model in eager mode.
      
      To support differences in TorchScript / eager mode outputs, an
      `unwrapper` function can be provided per-model.
      
      * Fix inception, use PYTORCH_TEST_WITH_SLOW
      
      * Update
      
      * Remove assertNestedTensorObjectsEqual
      
      * Add PYTORCH_TEST_WITH_SLOW to CircleCI config
      
      * Add MaskRCNN unwrapper
      
      * fix prec args
      
      * Remove CI changes
      
      * update
      
      * Update
      
      * remove expect changes
      
      * Fix tolerance bug
      
      * Fix breakages
      
      * Fix quantized resnet
      
      * Fix merge errors and simplify code
      
      * DeepLabV3 has been fixed
      
      * Temporarily disable jit compilation
      227027d5
  18. 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
  19. 21 Nov, 2019 1 commit
  20. 06 Nov, 2019 1 commit
  21. 28 Oct, 2019 1 commit
  22. 18 Oct, 2019 1 commit
  23. 15 Oct, 2019 1 commit
    • Lara Haidar's avatar
      Support Exporting RPN to ONNX (#1329) · 1d6145d1
      Lara Haidar authored
      * Support Exporting RPN to ONNX
      
      * address PR comments
      
      * fix cat
      
      * add flatten
      
      * replace cat by stack
      
      * update test to run only on rpn module
      
      * use tolerate_small_mismatch
      1d6145d1
  24. 18 Sep, 2019 1 commit
  25. 17 Sep, 2019 1 commit
  26. 30 Aug, 2019 1 commit
  27. 05 Aug, 2019 1 commit
  28. 23 Jul, 2019 1 commit
  29. 12 Jul, 2019 2 commits
  30. 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
  31. 04 Jul, 2019 2 commits
    • ekka's avatar
      Fix Mask R-CNN docs (#1089) · 9dfca9af
      ekka authored
      Fixes #1047.
      9dfca9af
    • buoyancy99's avatar
      fix transform for rcnns so original images list is unchanged (#1084) · e8e9bdb6
      buoyancy99 authored
      * fix transform for rcnns so original images are unchanged
      
      * transform does not change input list anymore
      
      transform does not change input list anymore. Improve code according to reviewer comment
      
      * transform for maskrcnn no longer modify input list
      
      transform for maskrcnn no longer modify input list. improve code according to comment
      
      * transform for maskrcnn no longer modifies input list
      e8e9bdb6
  32. 14 Jun, 2019 2 commits
  33. 03 Jun, 2019 1 commit
  34. 25 May, 2019 1 commit
  35. 22 May, 2019 1 commit