- 13 Oct, 2020 1 commit
-
-
Francisco Massa authored
* Add rough implementation of RetinaNet. * Move AnchorGenerator to a seperate file. * Move box similarity to Matcher. * Expose extra blocks in FPN. * Expose retinanet in __init__.py. * Use P6 and P7 in FPN for retinanet. * Use parameters from retinanet for anchor generation. * General fixes for retinanet model. * Implement loss for retinanet heads. * Output reshaped outputs from retinanet heads. * Add postprocessing of detections. * Small fixes. * Remove unused argument. * Remove python2 invocation of super. * Add postprocessing for additional outputs. * Add missing import of ImageList. * Remove redundant import. * Simplify class correction. * Fix pylint warnings. * Remove the label adjustment for background class. * Set default score threshold to 0.05. * Add weight initialization for regression layer. * Allow training on images with no annotations. * Use smooth_l1_loss with beta value. * Add more typehints for TorchScript conversions. * Fix linting issues. * Fix type hints in postprocess_detections. * Fix type annotations for TorchScript. * Fix inconsistency with matched_idxs. * Add retinanet model test. * Add missing JIT annotations. * Remove redundant model construction Make tests pass * Fix bugs during training on newer PyTorch and unused params in DDP Needs cleanup and to add back support for images with no annotations * Cleanup resnet_fpn_backbone * Use L1 loss for regression Gives 1mAP improvement over smooth l1 * Disable support for images with no annotations Need to fix distributed first * Fix retinanet tests Need to deduplicate those box checks * Fix Lint * Add pretrained model * Add training info for retinanet Co-authored-by:
Hans Gaiser <hansg91@gmail.com> Co-authored-by:
Hans Gaiser <hans.gaiser@robovalley.com> Co-authored-by:
Hans Gaiser <hans.gaiser@robohouse.com>
-
- 20 May, 2020 1 commit
-
-
Erik authored
* Update README.md added some clarity to get the examples executable. Waiting to hear back if instructions should mention to setup COCO dataset * Update README.md * Update README.md * Update README.md * Update README.md * Update README.md
-
- 26 Nov, 2019 1 commit
-
-
Yoshitomo Matsubara authored
-