1. 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
  2. 03 Jun, 2019 1 commit
  3. 22 May, 2019 1 commit
  4. 21 May, 2019 1 commit
  5. 20 May, 2019 2 commits
  6. 19 May, 2019 1 commit