- 19 Jan, 2021 1 commit
-
-
Vasilis Vryniotis authored
* Tag fasterrcnn mobilenetv3 model with 320, add new inference config that makes it 2x faster sacrificing a bit of mAP. * Add a high resolution fasterrcnn mobilenetv3 model. * Update tests and expected values.
-
- 18 Jan, 2021 1 commit
-
-
Vasilis Vryniotis authored
* Minor refactoring of a private method to make it reusuable. * Adding a FasterRCNN + MobileNetV3 with & w/o FPN models. * Reducing Resolution to 320-640 and anchor sizes to 16-256. * Increase anchor sizes. * Adding rpn score threshold param on the train script. * Adding trainable_backbone_layers param on the train script. * Adding rpn_score_thresh param directly in fasterrcnn_mobilenet_v3_large_fpn. * Remove fasterrcnn_mobilenet_v3_large prototype and update expected file. * Update documentation and adding weights. * Use buildin Identity. * Fix spelling.
-
- 14 Jan, 2021 1 commit
-
-
Vasilis Vryniotis authored
* Introduce small score threshold on rpn * Adding docs and fixing keypoint and mask. * Making value 0.0 by default for BC. * Fixing for onnx. * Update threshold. * Removing non-default threshold from reference scripts. Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
-
- 27 Nov, 2020 1 commit
-
-
Vasilis Vryniotis authored
* Enable support for images without annotations * Ensuring gradient propagates to RegressionHead. * Rewriting losses to remove branching. * Fix the seed on DeformConv autocast test.
-
- 07 Apr, 2020 1 commit
-
-
Francisco Massa authored
* Add tests for negative samples for Mask R-CNN and Keypoint R-CNN * Fix lint
-
- 20 Mar, 2020 1 commit
-
-
Monica Alfaro authored
* modified FasterRCNN to accept negative samples * remove debug lines * Change torch.zeros_like to torch.zerros * Add unit tests * take the `device` into account Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
-