1. 26 Nov, 2020 1 commit
  2. 20 Nov, 2020 2 commits
    • Vasilis Vryniotis's avatar
      Refactor & enable JIT tests in all models and add warnings if skipped (#3033) · 4521f6d1
      Vasilis Vryniotis authored
      * Enable jit tests in all models and add warning if checkModule() tests are skipped.
      
      * Turning on JIT tests on CI.
      
      * Fixing broken unit-tests.
      
      * Refactoring and cleaning up duplicate code.
      4521f6d1
    • Alexey Demyanchuk's avatar
      Add explicit check for number of channels (#3013) · a51c49e4
      Alexey Demyanchuk authored
      
      
      * Add explicit check for number of channels
      
      Example why you need to check it:
      `M = torch.randint(low=0, high=2, size=(6, 64, 64), dtype = torch.float)`
      When you put this input through to_pil_image without mode argument, it converts to uint8 here:
      ```
      if pic.is_floating_point() and mode != 'F':
                  pic = pic.mul(255).byte()
      ```
      and change the mode to RGB here:
      ```
      if mode is None and npimg.dtype == np.uint8:
                  mode = 'RGB'
      ```
      Image.fromarray doesn't raise if provided with mode RGB and just cut number of channels from what you have to 3
      
      * Check number of channels before processing
      
      * Add test for invalid number of channels
      
      * Add explicit check for number of channels
      
      Example why you need to check it:
      `M = torch.randint(low=0, high=2, size=(6, 64, 64), dtype = torch.float)`
      When you put this input through to_pil_image without mode argument, it converts to uint8 here:
      ```
      if pic.is_floating_point() and mode != 'F':
                  pic = pic.mul(255).byte()
      ```
      and change the mode to RGB here:
      ```
      if mode is None and npimg.dtype == np.uint8:
                  mode = 'RGB'
      ```
      Image.fromarray doesn't raise if provided with mode RGB and just cut number of channels from what you have to 3
      
      * Check number of channels before processing
      
      * Add test for invalid number of channels
      
      * Put check after channel dim unsqueeze
      
      * Add test if error message is matching
      
      * Delete redundant code
      
      * Bug fix in checking for bad types
      Co-authored-by: default avatarDemyanchuk <demyanca@mh-hannover.local>
      Co-authored-by: default avatarvfdev <vfdev.5@gmail.com>
      a51c49e4
  3. 19 Nov, 2020 2 commits
  4. 18 Nov, 2020 2 commits
    • Zhiqiang Wang's avatar
      Adds Anchor tests with ground-truth outputs (#2983) · b18a4757
      Zhiqiang Wang authored
      * Add AnchorGenerator with ground-truth outputs
      
      * Minor fixes
      b18a4757
    • Vasilis Vryniotis's avatar
      Support specifying output channels in io.image.read_image (#2988) · 4d6ba678
      Vasilis Vryniotis authored
      * Adding output channels implementation for pngs.
      
      * Adding tests for png.
      
      * Adding channels in the API and documentation.
      
      * Fixing formatting.
      
      * Refactoring test_image.py to remove huge grace_hopper_517x606.pth file from assets and reduce duplicate code. Moving jpeg assets used by encode and write unit-tests on their separate folders.
      
      * Adding output channels implementation for jpegs. Fix asset locations.
      
      * Add tests for JPEG, adding the channels in the API and documentation and adding checks for inputs.
      
      * Changing folder for unit-test.
      
      * Fixing windows flakiness, removing duplicate test.
      
      * Replacing components to channels.
      
      * Adding reference for supporting CMYK.
      
      * Minor changes: num_components to output_components, adding comments, fixing variable name etc.
      
      * Reverting output_components to num_components.
      
      * Replacing decoding with generic method on tests.
      
      * Palette converted to Gray.
      4d6ba678
  5. 12 Nov, 2020 2 commits
  6. 10 Nov, 2020 2 commits
  7. 09 Nov, 2020 5 commits
  8. 06 Nov, 2020 2 commits
    • vfdev's avatar
      House keeping improvements: (#2964) · f655e6a7
      vfdev authored
      - fixed problem with error computation between results
      - refactored tensor cast for resize
      - fixed round usage
      f655e6a7
    • Vasilis Vryniotis's avatar
      Fix flakiness on detection tests (#2966) · 7f7ff056
      Vasilis Vryniotis authored
      * Simplify the ACCEPT=True logic in assertExpected().
      
      * Separate the expected filename estimation from assertExpected
      
      * Unflatten expected values.
      
      * Assert for duplicate scores if primary check fails.
      
      * Remove custom exceptions for algorithms and add a compact function for shrinking large ouputs.
      
      * Removing unused variables.
      
      * Add warning and comments.
      
      * Re-enable all autocast unit-test for detection and marking the tests as skipped in partial validation.
      
      * Move test skip at the end.
      
      * Changing the warning message.
      7f7ff056
  9. 05 Nov, 2020 1 commit
  10. 04 Nov, 2020 2 commits
  11. 03 Nov, 2020 1 commit
  12. 30 Oct, 2020 2 commits
  13. 22 Oct, 2020 2 commits
  14. 21 Oct, 2020 3 commits
  15. 20 Oct, 2020 1 commit
  16. 16 Oct, 2020 2 commits
  17. 14 Oct, 2020 3 commits
  18. 13 Oct, 2020 2 commits
    • Bruno Korbar's avatar
      VideoAPI docs update (#2802) · 2831f11a
      Bruno Korbar authored
      
      
      * Video reader now returns dicts
      
      * docs update
      
      * Minor improvements
      Co-authored-by: default avatarBruno Korbar <bjuncek@Frazz.local>
      Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
      2831f11a
    • Francisco Massa's avatar
      RetinaNet object detection (take 2) (#2784) · 5bb81c8e
      Francisco Massa authored
      
      
      * Add rough implementation of RetinaNet.
      
      * Move AnchorGenerator to a seperate file.
      
      * Move box similarity to Matcher.
      
      * Expose extra blocks in FPN.
      
      * Expose retinanet in __init__.py.
      
      * Use P6 and P7 in FPN for retinanet.
      
      * Use parameters from retinanet for anchor generation.
      
      * General fixes for retinanet model.
      
      * Implement loss for retinanet heads.
      
      * Output reshaped outputs from retinanet heads.
      
      * Add postprocessing of detections.
      
      * Small fixes.
      
      * Remove unused argument.
      
      * Remove python2 invocation of super.
      
      * Add postprocessing for additional outputs.
      
      * Add missing import of ImageList.
      
      * Remove redundant import.
      
      * Simplify class correction.
      
      * Fix pylint warnings.
      
      * Remove the label adjustment for background class.
      
      * Set default score threshold to 0.05.
      
      * Add weight initialization for regression layer.
      
      * Allow training on images with no annotations.
      
      * Use smooth_l1_loss with beta value.
      
      * Add more typehints for TorchScript conversions.
      
      * Fix linting issues.
      
      * Fix type hints in postprocess_detections.
      
      * Fix type annotations for TorchScript.
      
      * Fix inconsistency with matched_idxs.
      
      * Add retinanet model test.
      
      * Add missing JIT annotations.
      
      * Remove redundant model construction
      
      Make tests pass
      
      * Fix bugs during training on newer PyTorch and unused params in DDP
      
      Needs cleanup and to add back support for images with no annotations
      
      * Cleanup resnet_fpn_backbone
      
      * Use L1 loss for regression
      
      Gives 1mAP improvement over smooth l1
      
      * Disable support for images with no annotations
      
      Need to fix distributed first
      
      * Fix retinanet tests
      
      Need to deduplicate those box checks
      
      * Fix Lint
      
      * Add pretrained model
      
      * Add training info for retinanet
      Co-authored-by: default avatarHans Gaiser <hansg91@gmail.com>
      Co-authored-by: default avatarHans Gaiser <hans.gaiser@robovalley.com>
      Co-authored-by: default avatarHans Gaiser <hans.gaiser@robohouse.com>
      5bb81c8e
  19. 12 Oct, 2020 1 commit
  20. 11 Oct, 2020 1 commit
  21. 09 Oct, 2020 1 commit
    • Bruno Korbar's avatar
      [documentation] video API documentation and wrapper (#2778) · d5379656
      Bruno Korbar authored
      
      
      * initial API documentation attempt
      
      * test the docs
      
      * initial commit
      
      * updating test to match the registration
      
      * adding the warning on unsucessful import
      
      * Try to do conditional import
      
      * Simple fix?
      
      * clearing up docs
      
      * docstring commit
      
      * Adding types in arguments
      Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
      
      * reverting warning commit
      
      * addressing Francisco's comments
      
      * Apply suggestions from code review
      Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
      
      * Revert "reverting warning commit"
      
      This reverts commit bd1a3dd4f3b97709ab59c744962e11174757f8ce.
      
      * Revert "adding the warning on unsucessful import"
      
      This reverts commit afef7df9eaa73bf80246e6d9114cb4c30b16f0ce.
      
      * remove warnings import
      Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
      d5379656