- 10 Jul, 2019 1 commit
-
-
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
-
- 03 Jun, 2019 1 commit
-
-
Gu-ni-kim authored
Add 'import torch' in example
-
- 22 May, 2019 1 commit
-
-
Francisco Massa authored
-
- 21 May, 2019 1 commit
-
-
Francisco Massa authored
-
- 20 May, 2019 2 commits
-
-
Francisco Massa authored
* Add more documentation for the ops * Add documentation for Faster R-CNN * Add documentation for Mask R-CNN and Keypoint R-CNN * Improve doc for RPN * Add basic doc for GeneralizedRCNNTransform * Lint fixes
-
Francisco Massa authored
* Add COCO pre-trained weights for Faster R-CNN R-50 FPN * Add weights for Mask R-CNN and Keypoint R-CNN
-
- 19 May, 2019 1 commit
-
-
Francisco Massa authored
* Split mask_rcnn.py into several files * Lint
-