1. 12 Mar, 2021 2 commits
  2. 11 Mar, 2021 2 commits
  3. 10 Mar, 2021 4 commits
  4. 08 Mar, 2021 2 commits
    • Nicolas Hug's avatar
      Speed up equalize transform: use bincount instead of histc (#3493) · 77e41870
      Nicolas Hug authored
      * use bincount instead of hist
      
      * only use bincount when on CPU
      
      * Added equality test for CPU vs cuda
      
      * Fix flake8 and tests
      
      * tuple instead of int for size
      77e41870
    • Nicolas Hug's avatar
      [OPS, IMP] New batched_nms implementation (#3426) · 414427dd
      Nicolas Hug authored
      
      
      * new batched_nms implem
      
      * flake8
      
      * hopefully fix torchscipt tests
      
      * Use where instead of nonzero
      
      * Use same threshold (4k) for CPU and GPU
      
      * Remove use of argsort
      
      * use views again
      
      * remove print
      
      * trying stuff, I don't know what's going on
      
      * previous passed onnx checks so the error isn't in _vanilla func. Trying to return vanilla now
      
      * add tracing decorators
      
      * cleanup
      
      * wip
      
      * ignore new path with ONNX
      
      * use vanilla if tracing...????
      
      * Remove script_if_tracing decorator as it was conflicting with _is_tracing
      
      * flake8
      
      * Improve coverage
      Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
      414427dd
  5. 04 Mar, 2021 3 commits
  6. 03 Mar, 2021 3 commits
  7. 02 Mar, 2021 2 commits
  8. 01 Mar, 2021 11 commits
  9. 26 Feb, 2021 1 commit
  10. 25 Feb, 2021 3 commits
  11. 23 Feb, 2021 2 commits
  12. 22 Feb, 2021 4 commits
  13. 18 Feb, 2021 1 commit
    • Philip Meier's avatar
      [POC] Base class for dataset tests (#3402) · 22c548b0
      Philip Meier authored
      * add base class for datasets tests
      
      * add better type hints
      
      * add documentation to subclasses
      
      * add utility functions to create files / folders of random images and videos
      
      * fix imports
      
      * remove class properties
      
      * fix smoke test
      
      * fix type hints
      
      * fix random size generation
      
      * add Caltech256 as example
      
      * add utility function to create grid of combinations
      
      * add CIFAR100? as example
      
      * lint
      
      * add missing import
      
      * improve documentation
      
      * create 1 frame videos by default
      
      * remove obsolete check
      
      * return path of files created with utility functions
      
      * [test] close PIL file handles before deletion
      
      * fix video folder creation
      
      * generalize file handle closing
      
      * fix lazy imports
      
      * add test for transforms
      
      * fix explanation comment
      
      * lint
      
      * force load opened PIL images
      
      * lint
      
      * copy default config to avoid inplace modification
      
      * enable additional arg forwarding
      22c548b0