Code

Add YOLOS#16848

Code

Merged

NielsRogge merged 30 commits intohuggingface:mainhuggingface/transformers:mainfrom NielsRogge:add_yolosNielsRogge/transformers:add_yolosCopy head branch name to clipboardMay 2, 2022

+3,608-60

Commits

Commits on May 2, 2022

• First draft

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  7a17bf6

  Copy full SHA for 7a17bf6

• Add YolosForObjectDetection

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  6b2e22e

  Copy full SHA for 6b2e22e

• Make forward pass work

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  4d77008

  Copy full SHA for 4d77008

• Add mid position embeddings

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  563be7b

  Copy full SHA for 563be7b

• Add interpolation of position encodings

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  c9aa75b

  Copy full SHA for c9aa75b

• Add expected values

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  f3aed34

  Copy full SHA for f3aed34

• Add YOLOS to tests

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  ec67166

  Copy full SHA for ec67166

• Add integration test

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  c8c80d9

  Copy full SHA for c8c80d9

• Support tiny model as well

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  d51c09a

  Copy full SHA for d51c09a

• Support all models in conversion script

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  bf7ad85

  Copy full SHA for bf7ad85

• Remove mid_pe_size attribute

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  a463318

  Copy full SHA for a463318

• Make more tests pass

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  39b478c

  Copy full SHA for 39b478c

• Add model to README and fix config

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  bf38802

  Copy full SHA for bf38802

• Add copied from statements

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  992de41

  Copy full SHA for 992de41

• Rename base_model_prefix to vit

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  f1460c3

  Copy full SHA for f1460c3

• Add missing YOLOS_PRETRAINED_CONFIG_ARCHIVE_MAP

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  86d6e34

  Copy full SHA for 86d6e34

• Apply suggestions from code review

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  74ca0c4

  Copy full SHA for 74ca0c4

• Apply more suggestions from code review

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  c03b734

  Copy full SHA for c03b734

• Convert remaining checkpoints

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  a7875e8

  Copy full SHA for a7875e8

• Improve docstrings

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  7c2dc6e

  Copy full SHA for 7c2dc6e

• Add YolosFeatureExtractor

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  2663957

  Copy full SHA for 2663957

• Add feature extractor to docs

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  a61626d

  Copy full SHA for a61626d

• Add corresponding tests

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  c31edd6

  Copy full SHA for c31edd6

• Fix style

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  450798d

  Copy full SHA for 450798d

• Fix docs

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  a72165f

  Copy full SHA for a72165f

• Apply suggestion from code review

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  dcc344a

  Copy full SHA for dcc344a

• Fix bad rebase

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  daa3a52

  Copy full SHA for daa3a52

• Fix some more bad rebase

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  d923bea

  Copy full SHA for d923bea

• Fix missing character

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  2567f4e

  Copy full SHA for 2567f4e

• Improve docs and variable names

  

  Niels Rogge

  authored and

  Niels Rogge

  committed
  a242ed1

  Copy full SHA for a242ed1