Code

Image Feature Extraction pipeline#28216

Code

Merged

amyeroberts merged 25 commits intohuggingface:mainhuggingface/transformers:mainfrom amyeroberts:image-fe-pipelineamyeroberts/transformers:image-fe-pipelineCopy head branch name to clipboardFeb 5, 2024

+387-53

Commits

Commits on Feb 5, 2024

• Draft pipeline

  amyeroberts

  committed
  12a4da1

  Copy full SHA for 12a4da1

• Fixup

  amyeroberts

  committed
  ec40e7d

  Copy full SHA for ec40e7d

• Fix docstrings

  amyeroberts

  committed
  26467fc

  Copy full SHA for 26467fc

• Update doctest

  amyeroberts

  committed
  1d0a6dd

  Copy full SHA for 1d0a6dd

• Update pipeline_model_mapping

  amyeroberts

  committed
  81303dc

  Copy full SHA for 81303dc

• Update docstring

  amyeroberts

  committed
  ba549d5

  Copy full SHA for ba549d5

• Update tests

  amyeroberts

  committed
  1468c54

  Copy full SHA for 1468c54

• Update src/transformers/pipelines/image_feature_extraction.py

  Show description for 102422b
  

  amyeroberts

  and

  osanseviero

  committed
  102422b

  Copy full SHA for 102422b

• Fix docstrings - review comments

  amyeroberts

  committed
  466ce4e

  Copy full SHA for 466ce4e

• Remove pipeline mapping for composite vision models

  amyeroberts

  committed
  1923eb1

  Copy full SHA for 1923eb1

• Add to pipeline tests

  amyeroberts

  committed
  9e598c9

  Copy full SHA for 9e598c9

• Remove for flava (multimodal)

  amyeroberts

  committed
  f990de9

  Copy full SHA for f990de9

• safe pil import

  amyeroberts

  committed
  999e1e1

  Copy full SHA for 999e1e1

• Add requirements for pipeline run

  amyeroberts

  committed
  3f85b37

  Copy full SHA for 3f85b37

• Account for super slow efficientnet

  amyeroberts

  committed
  960b74a

  Copy full SHA for 960b74a

• Review comments

  amyeroberts

  committed
  e4098cb

  Copy full SHA for e4098cb

• Fix tests

  amyeroberts

  committed
  50c0f63

  Copy full SHA for 50c0f63

• Swap order of kwargs

  amyeroberts

  committed
  42853ce

  Copy full SHA for 42853ce

• Use build_pipeline_init_args

  amyeroberts

  committed
  4b059f0

  Copy full SHA for 4b059f0

• Add back FE pipeline for Vilt

  amyeroberts

  committed
  2c5c2fb

  Copy full SHA for 2c5c2fb

• Include image_processor_kwargs in docstring

  amyeroberts

  committed
  95f9c58

  Copy full SHA for 95f9c58

• Mark test as flaky

  amyeroberts

  committed
  94945ec

  Copy full SHA for 94945ec

• Update TODO

  amyeroberts

  committed
  0a839a0

  Copy full SHA for 0a839a0

• Update tests/pipelines/test_pipelines_image_feature_extraction.py

  Show description for 3fd0797
  

  amyeroberts

  and

  ArthurZucker

  committed
  3fd0797

  Copy full SHA for 3fd0797

• Add license header

  amyeroberts

  committed
  04c109c

  Copy full SHA for 04c109c