Code

RetinaNet object detection (take 2)#2784

Code

Merged

fmassa merged 41 commits intopytorch:masterpytorch/vision:masterfrom fmassa:retinanetfmassa/vision-1:retinanetCopy head branch name to clipboardOct 13, 2020

+884-169

Commits

Commits on Oct 9, 2020

• Add rough implementation of RetinaNet.

  

  hgaiser

  authored and

  Hans Gaiser

  committed
  50f822c

  Copy full SHA for 50f822c

• Move AnchorGenerator to a seperate file.

  

  Hans Gaiser

  committed
  022f8e1

  Copy full SHA for 022f8e1

• Move box similarity to Matcher.

  

  hgaiser

  authored and

  Hans Gaiser

  committed
  8e0804d

  Copy full SHA for 8e0804d

• Expose extra blocks in FPN.

  

  hgaiser

  authored and

  Hans Gaiser

  committed
  ad53194

  Copy full SHA for ad53194

• Expose retinanet in __init__.py.

  

  hgaiser

  authored and

  Hans Gaiser

  committed
  2a5a5be

  Copy full SHA for 2a5a5be

• Use P6 and P7 in FPN for retinanet.

  

  hgaiser

  authored and

  Hans Gaiser

  committed
  49e990c

  Copy full SHA for 49e990c

• Use parameters from retinanet for anchor generation.

  

  hgaiser

  authored and

  Hans Gaiser

  committed
  b5966eb

  Copy full SHA for b5966eb

• General fixes for retinanet model.

  

  hgaiser

  authored and

  Hans Gaiser

  committed
  aab1b28

  Copy full SHA for aab1b28

• Implement loss for retinanet heads.

  

  hgaiser

  authored and

  Hans Gaiser

  committed
  c078114

  Copy full SHA for c078114

• Output reshaped outputs from retinanet heads.

  

  hgaiser

  authored and

  Hans Gaiser

  committed
  eae4ee5

  Copy full SHA for eae4ee5

• Add postprocessing of detections.

  

  hgaiser

  authored and

  Hans Gaiser

  committed
  3dac477

  Copy full SHA for 3dac477

• Small fixes.

  

  hgaiser

  authored and

  Hans Gaiser

  committed
  9981a3c

  Copy full SHA for 9981a3c

• Remove unused argument.

  

  hgaiser

  authored and

  Hans Gaiser

  committed
  5571dfe

  Copy full SHA for 5571dfe

• Remove python2 invocation of super.

  

  hgaiser

  authored and

  Hans Gaiser

  committed
  fc7751b

  Copy full SHA for fc7751b

• Add postprocessing for additional outputs.

  

  hgaiser

  authored and

  Hans Gaiser

  committed
  b942648

  Copy full SHA for b942648

• Add missing import of ImageList.

  

  hgaiser

  authored and

  Hans Gaiser

  committed
  b619936

  Copy full SHA for b619936

• Remove redundant import.

  

  hgaiser

  authored and

  Hans Gaiser

  committed
  8c86588

  Copy full SHA for 8c86588

• Simplify class correction.

  

  hgaiser

  authored and

  Hans Gaiser

  committed
  2934f0d

  Copy full SHA for 2934f0d

• Fix pylint warnings.

  

  hgaiser

  authored and

  Hans Gaiser

  committed
  32b8e77

  Copy full SHA for 32b8e77

• Remove the label adjustment for background class.

  

  Hans Gaiser

  authored and

  Hans Gaiser

  committed
  437bfe9

  Copy full SHA for 437bfe9

• Set default score threshold to 0.05.

  

  Hans Gaiser

  authored and

  Hans Gaiser

  committed
  9e810d6

  Copy full SHA for 9e810d6

• Add weight initialization for regression layer.

  

  Hans Gaiser

  authored and

  Hans Gaiser

  committed
  f7d8c2e

  Copy full SHA for f7d8c2e

• Allow training on images with no annotations.

  

  Hans Gaiser

  authored and

  Hans Gaiser

  committed
  d86c437

  Copy full SHA for d86c437

• Use smooth_l1_loss with beta value.

  

  Hans Gaiser

  committed
  72e46f2

  Copy full SHA for 72e46f2

• Add more typehints for TorchScript conversions.

  

  hgaiser

  authored and

  Hans Gaiser

  committed
  41c90fa

  Copy full SHA for 41c90fa

• Fix linting issues.

  

  hgaiser

  authored and

  Hans Gaiser

  committed
  b9daa86

  Copy full SHA for b9daa86

• Fix type hints in postprocess_detections.

  

  hgaiser

  authored and

  Hans Gaiser

  committed
  97d63b6

  Copy full SHA for 97d63b6

• Fix type annotations for TorchScript.

  

  Hans Gaiser

  committed
  eba7e16

  Copy full SHA for eba7e16

• Fix inconsistency with matched_idxs.

  

  Hans Gaiser

  committed
  9545059

  Copy full SHA for 9545059

• Add retinanet model test.

  

  Hans Gaiser

  committed
  4865952

  Copy full SHA for 4865952

• Add missing JIT annotations.

  

  Hans Gaiser

  committed
  6e065be

  Copy full SHA for 6e065be

Commits on Oct 10, 2020

• Remove redundant model construction

  Show description for 7dc4c6b
  fmassa

  committed
  7dc4c6b

  Copy full SHA for 7dc4c6b

Commits on Oct 11, 2020

• Fix bugs during training on newer PyTorch and unused params in DDP

  Show description for 640e59b
  fmassa

  committed
  640e59b

  Copy full SHA for 640e59b

• Cleanup resnet_fpn_backbone

  fmassa

  committed
  23cabe3

  Copy full SHA for 23cabe3

Commits on Oct 13, 2020

• Use L1 loss for regression

  Show description for 214ead7
  fmassa

  committed
  214ead7

  Copy full SHA for 214ead7

• Disable support for images with no annotations

  Show description for 44a1333
  fmassa

  committed
  44a1333

  Copy full SHA for 44a1333

• Merge branch 'master' of github.com:pytorch/vision into retinanet

  fmassa

  committed
  c2f6334

  Copy full SHA for c2f6334

• Fix retinanet tests

  Show description for e560039
  fmassa

  committed
  e560039

  Copy full SHA for e560039

• Fix Lint

  fmassa

  committed
  0647732

  Copy full SHA for 0647732

• Add pretrained model

  fmassa

  committed
  aa6364f

  Copy full SHA for aa6364f

• Add training info for retinanet

  fmassa

  committed
  9b62169

  Copy full SHA for 9b62169