Code

Add Mask R-CNN#25348

Code

Closed

NielsRogge wants to merge 147 commits intohuggingface:mainhuggingface/transformers:mainfrom NielsRogge:add_maskrcnn_23_07NielsRogge/transformers:add_maskrcnn_23_07Copy head branch name to clipboard

+6,239-3

Commits

Commits on Oct 16, 2023

• First draft

  NielsRogge

  committed
  27060b3

  Copy full SHA for 27060b3

• Fix conversion script for backbone

  NielsRogge

  committed
  628daaa

  Copy full SHA for 628daaa

• More progress

  NielsRogge

  committed
  a757379

  Copy full SHA for a757379

• More improvements

  NielsRogge

  committed
  32ffd9a

  Copy full SHA for 32ffd9a

• Fix conversion without layernorm

  NielsRogge

  committed
  604913e

  Copy full SHA for 604913e

• More fixes

  NielsRogge

  committed
  bbc5505

  Copy full SHA for bbc5505

• More fixes

  NielsRogge

  committed
  5c8cc73

  Copy full SHA for 5c8cc73

• More improvements

  NielsRogge

  committed
  feb9cf8

  Copy full SHA for feb9cf8

• More improvements

  NielsRogge

  committed
  0e6e936

  Copy full SHA for 0e6e936

• Use torchvision's nms implementation

  NielsRogge

  committed
  4d88816

  Copy full SHA for 4d88816

• Add initial draft of RoI head

  NielsRogge

  committed
  0527e26

  Copy full SHA for 0527e26

• Make bounding box results match original implementation

  NielsRogge

  committed
  f856b80

  Copy full SHA for f856b80

• Add mask head

  NielsRogge

  committed
  5a324c9

  Copy full SHA for 5a324c9

• Verify mask head

  NielsRogge

  committed
  71c59fe

  Copy full SHA for 71c59fe

• Add Assigner and BboxOverlaps classes

  NielsRogge

  committed
  0ee9849

  Copy full SHA for 0ee9849

• Add sampler

  NielsRogge

  committed
  aa6c7e5

  Copy full SHA for aa6c7e5

• Improve forward of RPN

  NielsRogge

  committed
  09c34f3

  Copy full SHA for 09c34f3

• Add training support for RPN

  NielsRogge

  committed
  4639585

  Copy full SHA for 4639585

• More improvements

  NielsRogge

  committed
  2b049a5

  Copy full SHA for 2b049a5

• Add forward_train to roi head

  NielsRogge

  committed
  19ddeb1

  Copy full SHA for 19ddeb1

• Add loss calculation of mask head

  NielsRogge

  committed
  48b30d0

  Copy full SHA for 48b30d0

• Add loss computation of mask head

  NielsRogge

  committed
  f91eb92

  Copy full SHA for f91eb92

• Improve docstrings

  NielsRogge

  committed
  e50f2f0

  Copy full SHA for e50f2f0

• Add losses to output

  NielsRogge

  committed
  7740932

  Copy full SHA for 7740932

• Add cpu method

  NielsRogge

  committed
  f768d3b

  Copy full SHA for f768d3b

• Remove checké

  NielsRogge

  committed
  e7cd695

  Copy full SHA for e7cd695

• Deactive mask loss

  NielsRogge

  committed
  a034585

  Copy full SHA for a034585

• Activate mask loss

  NielsRogge

  committed
  5c044b1

  Copy full SHA for 5c044b1

• Remove img_metas default value

  NielsRogge

  committed
  1118b38

  Copy full SHA for 1118b38

• Add integration test, general fixes

  NielsRogge

  committed
  49ea99a

  Copy full SHA for 49ea99a

• Delete scripts and add integration test for training

  NielsRogge

  committed
  35e894a

  Copy full SHA for 35e894a

• Fix training integration test

  NielsRogge

  committed
  0cfd7c7

  Copy full SHA for 0cfd7c7

• Apply suggestions from code review

  NielsRogge

  committed
  2d14522

  Copy full SHA for 2d14522

• Apply some more suggestions

  NielsRogge

  committed
  86250c4

  Copy full SHA for 86250c4

• Fix docstrings, remove multi_apply, update integration test

  NielsRogge

  committed
  82d8e66

  Copy full SHA for 82d8e66

• Update integration test

  NielsRogge

  committed
  2169920

  Copy full SHA for 2169920

• Fix docstrings

  NielsRogge

  committed
  3e868e8

  Copy full SHA for 3e868e8

• Simplify RPN forward

  NielsRogge

  committed
  e5c90ae

  Copy full SHA for e5c90ae

• Comment out most print statements

  NielsRogge

  committed
  26adbc1

  Copy full SHA for 26adbc1

• Add print statements

  NielsRogge

  committed
  077d4f6

  Copy full SHA for 077d4f6

• Update README, remove multi_apply

  NielsRogge

  committed
  08b89b4

  Copy full SHA for 08b89b4

• Remove multi_apply

  NielsRogge

  committed
  0c4b7b5

  Copy full SHA for 0c4b7b5

• Switch to channels first for img_metas

  NielsRogge

  committed
  86be4d0

  Copy full SHA for 86be4d0

• More improvements

  NielsRogge

  committed
  fac7457

  Copy full SHA for fac7457

• Add post_process method

  NielsRogge

  committed
  e652344

  Copy full SHA for e652344

• Make feature extractor importable

  NielsRogge

  committed
  3db4a6b

  Copy full SHA for 3db4a6b

• Add COCO API

  NielsRogge

  committed
  0820b45

  Copy full SHA for 0820b45

• Add instance segmentation postprocessing

  NielsRogge

  committed
  0ac1ff0

  Copy full SHA for 0ac1ff0

• Add fix

  NielsRogge

  committed
  e640e7f

  Copy full SHA for e640e7f

• Apply another fix

  NielsRogge

  committed
  11689ec

  Copy full SHA for 11689ec

• Add get_seg_masks to feature extractor

  NielsRogge

  committed
  5a5c4c0

  Copy full SHA for 5a5c4c0

• Add attribute

  NielsRogge

  committed
  599614d

  Copy full SHA for 599614d

• Fix docs

  NielsRogge

  committed
  78939c4

  Copy full SHA for 78939c4

• Add FPN hidden states to outputs

  NielsRogge

  committed
  29a4fb8

  Copy full SHA for 29a4fb8

• Update integration test

  NielsRogge

  committed
  b20085e

  Copy full SHA for b20085e

• Improve integration test

  NielsRogge

  committed
  5f707d9

  Copy full SHA for 5f707d9

• Improve postprocessing methods

  NielsRogge

  committed
  78ea18b

  Copy full SHA for 78ea18b

• Improve postprocessing

  NielsRogge

  committed
  84d96b2

  Copy full SHA for 84d96b2

• Rescale by default

  NielsRogge

  committed
  bff8283

  Copy full SHA for bff8283

• Rescale by default

  NielsRogge

  committed
  39dc2d0

  Copy full SHA for 39dc2d0

• Minor fix

  NielsRogge

  committed
  d451531

  Copy full SHA for d451531

• Create nms and loss_utils files

  NielsRogge

  committed
  2db237b

  Copy full SHA for 2db237b

• Improve docstrings and fix tests

  NielsRogge

  committed
  5f5e5ea

  Copy full SHA for 5f5e5ea

• Rename model to MaskRCNN

  NielsRogge

  committed
  7cf45cc

  Copy full SHA for 7cf45cc

• Fix conversion script

  NielsRogge

  committed
  f6e652a

  Copy full SHA for f6e652a

• Use AutoBackbone

  NielsRogge

  committed
  d044dae

  Copy full SHA for d044dae

• Fix integration tests

  NielsRogge

  committed
  2ca7934

  Copy full SHA for 2ca7934

• Remove ori_shape

  NielsRogge

  committed
  0a524da

  Copy full SHA for 0a524da

• More fixes

  NielsRogge

  committed
  3d15c65

  Copy full SHA for 3d15c65

• Update model outputs

  NielsRogge

  committed
  06fa085

  Copy full SHA for 06fa085

• Update model outputs

  NielsRogge

  committed
  e517a23

  Copy full SHA for e517a23

• Fix more tests

  NielsRogge

  committed
  86cc77d

  Copy full SHA for 86cc77d

• Add image processor

  NielsRogge

  committed
  734db05

  Copy full SHA for 734db05

• Remove torchvision from test file

  NielsRogge

  committed
  3b406e6

  Copy full SHA for 3b406e6

• Add img_metas

  NielsRogge

  committed
  ff4ab2f

  Copy full SHA for ff4ab2f

• Fix rebase

  NielsRogge

  committed
  969436a

  Copy full SHA for 969436a

• Run fix-copies

  NielsRogge

  committed
  d2dfc2c

  Copy full SHA for d2dfc2c

• Fix docs

  NielsRogge

  committed
  edc9253

  Copy full SHA for edc9253

• Add annotations to image processor

  NielsRogge

  committed
  a401e70

  Copy full SHA for a401e70

• Improve image processor, add tests

  NielsRogge

  committed
  ee2f6fc

  Copy full SHA for ee2f6fc

• Fix style

  NielsRogge

  committed
  d69403f

  Copy full SHA for d69403f

• Remove img_metas

  NielsRogge

  committed
  b8d6573

  Copy full SHA for b8d6573

• Add pipeline support

  NielsRogge

  committed
  e2bf821

  Copy full SHA for e2bf821

• More improvements

  NielsRogge

  committed
  91b2824

  Copy full SHA for 91b2824

• More improvements

  NielsRogge

  committed
  c3d5bb3

  Copy full SHA for c3d5bb3

• More improvements

  NielsRogge

  committed
  0511012

  Copy full SHA for 0511012

• Add docs

  NielsRogge

  committed
  c7ddb5f

  Copy full SHA for c7ddb5f

• Update READMEs

  NielsRogge

  committed
  0f55201

  Copy full SHA for 0f55201

• Undo changes to convnext models

  NielsRogge

  committed
  a0058cd

  Copy full SHA for a0058cd

• Delete files

  NielsRogge

  committed
  50671ef

  Copy full SHA for 50671ef

• Delete another file

  NielsRogge

  committed
  e1a3afb

  Copy full SHA for e1a3afb

• Fix blank lines

  NielsRogge

  committed
  5e985a0

  Copy full SHA for 5e985a0

• Fix toctree, use is_torchvision_available

  NielsRogge

  committed
  fd479f8

  Copy full SHA for fd479f8

• More fixes

  NielsRogge

  committed
  328fa8c

  Copy full SHA for 328fa8c

• Fix another import

  NielsRogge

  committed
  ae1fce9

  Copy full SHA for ae1fce9

• More fixes

  NielsRogge

  committed
  87bddb1

  Copy full SHA for 87bddb1

• Improve docstrings and variable names

  NielsRogge

  committed
  a70cc3a

  Copy full SHA for a70cc3a

• Improve docstrings

  NielsRogge

  committed
  c9d6ef6

  Copy full SHA for c9d6ef6

• More improvements

  NielsRogge

  committed
  10a67fa

  Copy full SHA for 10a67fa

• Fix more docstrings

  NielsRogge

  committed
  e6dd8f7

  Copy full SHA for e6dd8f7

• Improve docstrings

  NielsRogge

  committed
  5ecdb35

  Copy full SHA for 5ecdb35

• More improvements

  NielsRogge

  committed
  cbf3a04

  Copy full SHA for cbf3a04

• Move utils to model folder

  NielsRogge

  committed
  4611a56

  Copy full SHA for 4611a56

• Fix docstrings

  NielsRogge

  committed
  3e27dee

  Copy full SHA for 3e27dee

• More improvements

  NielsRogge

  committed
  b57fb2c

  Copy full SHA for b57fb2c

• Move SamplingResult and MaskTarget into the modeling file

  NielsRogge

  committed
  1dacba3

  Copy full SHA for 1dacba3

• Add more docstrings, simplify l1 loss

  NielsRogge

  committed
  b1620e3

  Copy full SHA for b1620e3

• More improvements

  NielsRogge

  committed
  c55d3f4

  Copy full SHA for c55d3f4

• More improvements

  NielsRogge

  committed
  6b677ad

  Copy full SHA for 6b677ad

• Remove unmap, improve docstrings

  NielsRogge

  committed
  8710bfb

  Copy full SHA for 8710bfb

• Add unmap

  NielsRogge

  committed
  a39e991

  Copy full SHA for a39e991

• More improvements

  NielsRogge

  committed
  1cce572

  Copy full SHA for 1cce572

• Add more docstrings

  NielsRogge

  committed
  307d8f4

  Copy full SHA for 307d8f4

• More improvements

  NielsRogge

  committed
  dffa15d

  Copy full SHA for dffa15d

• Remove unused arguments

  NielsRogge

  committed
  a56a4e1

  Copy full SHA for a56a4e1

• Move NMS into the modeling code

  NielsRogge

  committed
  910f7db

  Copy full SHA for 910f7db

• Move loss utils in the modeling file

  NielsRogge

  committed
  c0e60e5

  Copy full SHA for c0e60e5

• Remove vilt

  NielsRogge

  committed
  99b7e61

  Copy full SHA for 99b7e61

• Remove markdown file

  NielsRogge

  committed
  d285036

  Copy full SHA for d285036

• Use normalize of base image processor

  NielsRogge

  committed
  85105e3

  Copy full SHA for 85105e3

• Address comments

  NielsRogge

  committed
  cb1e88e

  Copy full SHA for cb1e88e

• Address more comments

  NielsRogge

  committed
  a89fb26

  Copy full SHA for a89fb26

• Address more comments

  NielsRogge

  committed
  45261f4

  Copy full SHA for 45261f4

• Improve object detection pipeline

  NielsRogge

  committed
  4c7894b

  Copy full SHA for 4c7894b

• Fix pipeline

  NielsRogge

  committed
  aa0db1d

  Copy full SHA for aa0db1d

• Minor fix

  NielsRogge

  committed
  d7fd33a

  Copy full SHA for d7fd33a

• Always return indices, improve image processor tests

  NielsRogge

  committed
  5cfe97c

  Copy full SHA for 5cfe97c

• Address comment

  NielsRogge

  committed
  ee9bd2d

  Copy full SHA for ee9bd2d

• Address more comments

  NielsRogge

  committed
  f2969b4

  Copy full SHA for f2969b4

• Address comments

  NielsRogge

  committed
  bf2171d

  Copy full SHA for bf2171d

• Address more comments, fix copies

  NielsRogge

  committed
  226fc49

  Copy full SHA for 226fc49

• Move NMS code to image processor

  NielsRogge

  committed
  87fb254

  Copy full SHA for 87fb254

• Move bbox coder to separate file

  NielsRogge

  committed
  614ef8b

  Copy full SHA for 614ef8b

• Improve variable names

  NielsRogge

  committed
  70f21c0

  Copy full SHA for 70f21c0

• Address more comments

  NielsRogge

  committed
  a506b35

  Copy full SHA for a506b35

• More improvements

  NielsRogge

  committed
  ea11991

  Copy full SHA for ea11991

• Replace lists by tuples

  NielsRogge

  committed
  6b2bf6c

  Copy full SHA for 6b2bf6c

• Improve docstrings

  NielsRogge

  committed
  8553218

  Copy full SHA for 8553218

• Remove function

  NielsRogge

  committed
  9663616

  Copy full SHA for 9663616

• Improve docstrings

  NielsRogge

  committed
  ee17796

  Copy full SHA for ee17796

• Improve image processor

  NielsRogge

  committed
  e251688

  Copy full SHA for e251688

• More improvements

  NielsRogge

  committed
  d3e934c

  Copy full SHA for d3e934c

• Add docstring

  NielsRogge

  committed
  3d769a6

  Copy full SHA for 3d769a6

• Remove unused code

  NielsRogge

  committed
  fea1f4b

  Copy full SHA for fea1f4b

• Rename function

  NielsRogge

  committed
  8268145

  Copy full SHA for 8268145

• Improve docstrings, variable names

  NielsRogge

  committed
  901d7f9

  Copy full SHA for 901d7f9

• Improve docstrings

  NielsRogge

  committed
  0250361

  Copy full SHA for 0250361