1. 21 Apr, 2021 1 commit
  2. 20 Apr, 2021 1 commit
  3. 19 Apr, 2021 1 commit
  4. 16 Apr, 2021 1 commit
  5. 12 Apr, 2021 2 commits
  6. 09 Apr, 2021 1 commit
    • Prabhat Roy's avatar
      Added KITTI dataset (#3640) · 7da9afee
      Prabhat Roy authored
      
      
      * Added KITTI dataset
      
      * Addressed review comments
      
      * Changed type of target to List[Dict] and corrected the data types of the returned values.
      
      * Updated unit test to rely on ImageDatasetTestCase
      
      * Added kitti to dataset documentation
      
      * Cleaned up test and some minor changes
      
      * Made data_url a string instead of a list
      
      * Removed unnecessary try and print
      Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
      7da9afee
  7. 08 Apr, 2021 1 commit
    • Nicolas Hug's avatar
      Add Quantized version of RoIAlign (#3624) · ad9cc62a
      Nicolas Hug authored
      * WIP
      
      * clang
      
      * docs
      
      * extracted out common utils
      
      * Use better quantization function and pass tensors as parameters
      
      * proper dequantization
      
      * Some tests
      
      * Dequantization optimization, seems to gain a few ms
      
      * clang-format
      
      * again
      
      * more correct test. Had to remove optimization although it almost works
      
      * Also test aligned=True
      
      * remove useless part
      
      * more docs and comments
      
      * Put back optimization with more robust test
      
      * Added check for index upper bound
      
      * avoid possible overflow
      
      * Move common function into common.h
      
      * oops
      
      * scale=1,zero_point=0 makes more sense
      
      * Force batch size of 1 to prevent any indexingbug
      
      * format
      
      * format again
      
      * updated docstring
      
      * put back description comment for pre_calc_bilinear_interpolate
      
      * revert most changes to docstring as it's taken care of in another PR
      ad9cc62a
  8. 06 Apr, 2021 1 commit
  9. 30 Mar, 2021 5 commits
  10. 25 Mar, 2021 1 commit
  11. 24 Mar, 2021 2 commits
    • Philip Meier's avatar
      add tests for MNIST and variants (#3423) · aa365993
      Philip Meier authored
      * add tests for MNIST and variants
      
      * remove old tests and fakedata generation
      
      * fix default config detection for if dataset has variable keywords
      
      * use split="mnist" as default for EMNIST
      
      * fix QMNIST tests
      
      * lint
      
      * fix special kwargs detection
      
      * Revert "use split="mnist" as default for EMNIST"
      
      This reverts commit 62c9b23597f4a391cad409cbd93edac1b3474acf.
      
      * fix tests
      
      * fix QMNIST test case name
      
      * remove dead code from test
      
      * Revert "remove old tests and fakedata generation"
      
      This reverts commit a285b97c4827566a5bc06c288f5bba8d614a4f7a.
      
      * remove old tests
      
      * readd removed import
      aa365993
    • Philip Meier's avatar
      Improve error handling in make_dataset (#3496) · 0818c682
      Philip Meier authored
      * factor out find_classes
      
      * use find_classes in video datasets
      
      * adapt old tests
      0818c682
  12. 22 Mar, 2021 2 commits
  13. 20 Mar, 2021 1 commit
  14. 19 Mar, 2021 1 commit
  15. 17 Mar, 2021 2 commits
  16. 16 Mar, 2021 2 commits
  17. 15 Mar, 2021 2 commits
  18. 12 Mar, 2021 3 commits
  19. 11 Mar, 2021 2 commits
  20. 10 Mar, 2021 4 commits
  21. 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
  22. 04 Mar, 2021 2 commits