• 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
faster_rcnn.py 15.5 KB