RetinaNet object detection (take 2)#2784

Merged

fmassa merged 41 commits intopytorch:masterfrom

fmassa:retinanet

Oct 13, 2020

Commits on Oct 9, 2020

Add rough implementation of RetinaNet.

hgaiser
authored and
Hans Gaiser
committed
Move AnchorGenerator to a seperate file.
Hans Gaiser
committed
Move box similarity to Matcher.

hgaiser
authored and
Hans Gaiser
committed
Expose extra blocks in FPN.

hgaiser
authored and
Hans Gaiser
committed
Expose retinanet in __init__.py.

hgaiser
authored and
Hans Gaiser
committed
Use P6 and P7 in FPN for retinanet.

hgaiser
authored and
Hans Gaiser
committed
Use parameters from retinanet for anchor generation.

hgaiser
authored and
Hans Gaiser
committed
General fixes for retinanet model.

hgaiser
authored and
Hans Gaiser
committed
Implement loss for retinanet heads.

hgaiser
authored and
Hans Gaiser
committed
Output reshaped outputs from retinanet heads.

hgaiser
authored and
Hans Gaiser
committed
Add postprocessing of detections.

hgaiser
authored and
Hans Gaiser
committed
Small fixes.

hgaiser
authored and
Hans Gaiser
committed
Remove unused argument.

hgaiser
authored and
Hans Gaiser
committed
Remove python2 invocation of super.

hgaiser
authored and
Hans Gaiser
committed
Add postprocessing for additional outputs.

hgaiser
authored and
Hans Gaiser
committed
Add missing import of ImageList.

hgaiser
authored and
Hans Gaiser
committed
Remove redundant import.

hgaiser
authored and
Hans Gaiser
committed
Simplify class correction.

hgaiser
authored and
Hans Gaiser
committed
Fix pylint warnings.

hgaiser
authored and
Hans Gaiser
committed
Remove the label adjustment for background class.

Hans Gaiser
authored and
Hans Gaiser
committed
Set default score threshold to 0.05.

Hans Gaiser
authored and
Hans Gaiser
committed
Add weight initialization for regression layer.

Hans Gaiser
authored and
Hans Gaiser
committed
Allow training on images with no annotations.

Hans Gaiser
authored and
Hans Gaiser
committed
Use smooth_l1_loss with beta value.
Hans Gaiser
committed
Add more typehints for TorchScript conversions.

hgaiser
authored and
Hans Gaiser
committed
Fix linting issues.

hgaiser
authored and
Hans Gaiser
committed
Fix type hints in postprocess_detections.

hgaiser
authored and
Hans Gaiser
committed
Fix type annotations for TorchScript.
Hans Gaiser
committed
Fix inconsistency with matched_idxs.
Hans Gaiser
committed
Add retinanet model test.
Hans Gaiser
committed
Add missing JIT annotations.
Hans Gaiser
committed

Commits on Oct 10, 2020

Remove redundant model construction
fmassa
committed

Commits on Oct 11, 2020