1. 27 Jan, 2020 1 commit
    • Jon Crall's avatar
      Enhance AssignResult and SamplingResult (#1995) · 8457bbab
      Jon Crall authored
      * Enhance AssignResult and SamplingResult
      
      Add runtime dependency on ubelt (pending approval)
      
      Fix issue in SamplingResult.__init__
      
      Add rng as attribute of RandomSampler
      
      * fix linters
      
      * remove ubelt
      
      * Fix linters
      
      * fix linters again
      8457bbab
  2. 24 Dec, 2019 1 commit
    • Jon Crall's avatar
      Allow for images to contain zero true detections (#1531) · b6966700
      Jon Crall authored
      
      
      * Allow for images to contain zero true detections
      
      * Allow for empty assignment in PointAssigner
      
      * Allow ApproxMaxIouAssigner to return an empty result
      
      * Fix CascadeRNN forward when entire batch has no truth
      
      * Correctly assign boxes to background when there is no truth
      
      * Fix assignment tests
      
      * Make flatten robust
      
      * Fix bbox loss with empty pred/truth
      
      * Fix logic error in BBoxHead.loss
      
      * Add tests for empty truth cases
      
      * tests faster rcnn empty forward
      
      * Skip roipool forward tests if torchvision is not installed
      
      * Add tests for bbox/anchor heads
      
      * Consolidate test_forward and test_forward2
      
      * Fix assign_results.labels = None when gt_labels is given; Add test for this case
      
      * Fix OHEM Sampler with zero truth
      
      * remove xdev
      
      * resolve 3 reviews
      
      * Fix flake8
      
      * refactoring
      
      * fix yaml format
      
      * add filter flag
      
      * minor fix
      
      * delete redundant code in load anno
      
      * fix flake8 errors
      
      * quick fix for empty truth with masks
      
      * fix yapf error
      
      * fix mask padding for empty masks
      Co-authored-by: default avatarCao Yuhang <yhcao6@gmail.com>
      Co-authored-by: default avatarKai Chen <chenkaidev@gmail.com>
      b6966700