- 02 Jan, 2020 2 commits
-
-
Francisco Massa authored
* Fix lint following #1695 * V2 * V3
-
Prajjwal Bhargava authored
-
- 25 Nov, 2019 1 commit
-
-
eellison authored
* almost working... * respond to comments * add empty tensor op, handle different output types in generalized rcnn * clean ups * address comments * more changes * it's working! * torchscript bugs * add script/ eager test * eval script model * fix flake * division import * py2 compat * update test, fix arange bug * import division statement * fix linter * fixes * changes needed for JIT master * cleanups * remove imagelist_to * requested changes * Make FPN backwards-compatible and torchscript compatible We remove support for feature channels=0, but support for it was already a bit limited * Fix ONNX regression
-
- 12 Jul, 2019 1 commit
-
-
Varun Agrawal authored
updated all docstrings and code references for boxes to be consistent with the scheme (x1, y1, x2, y2) (#1110)
-
- 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
-