Skip to content

Commit 705588e

Browse files
merveenoyanMishig
andauthored
Add Image feature extraction on Hub as a task (#402)
Co-authored-by: Mishig <[email protected]>
1 parent 8d3d358 commit 705588e

File tree

4 files changed

+31
-0
lines changed

4 files changed

+31
-0
lines changed

packages/tasks/src/pipelines.ts

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -651,6 +651,11 @@ export const PIPELINE_DATA = {
651651
modality: "cv",
652652
color: "green",
653653
},
654+
"image-feature-extraction": {
655+
name: "Image Feature Extraction",
656+
modality: "cv",
657+
color: "indigo",
658+
},
654659
other: {
655660
name: "Other",
656661
modality: "other",

packages/tasks/src/tasks/index.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,7 @@ export const TASKS_MODEL_LIBRARIES: Record<PipelineType, ModelLibraryKey[]> = {
5252
"fill-mask": ["transformers", "transformers.js"],
5353
"graph-ml": ["transformers"],
5454
"image-classification": ["keras", "timm", "transformers", "transformers.js"],
55+
"image-feature-extraction": ["timm", "transformers"],
5556
"image-segmentation": ["transformers", "transformers.js"],
5657
"image-text-to-text": ["transformers"],
5758
"image-to-image": ["diffusers", "transformers", "transformers.js"],
@@ -169,6 +170,7 @@ export const TASKS_DATA: Record<PipelineType, TaskData | undefined> = {
169170
"zero-shot-object-detection": getData("zero-shot-object-detection", zeroShotObjectDetection),
170171
"text-to-3d": getData("text-to-3d", placeholder),
171172
"image-to-3d": getData("image-to-3d", placeholder),
173+
"image-feature-extraction": getData("image-feature-extraction", placeholder),
172174
} as const;
173175

174176
export interface ExampleRepo {
Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
1+
<script lang="ts">
2+
export let classNames = "";
3+
</script>
4+
5+
<svg
6+
class={classNames}
7+
xmlns="http://www.w3.org/2000/svg"
8+
xmlns:xlink="http://www.w3.org/1999/xlink"
9+
aria-hidden="true"
10+
fill="currentColor"
11+
focusable="false"
12+
role="img"
13+
width="1em"
14+
height="1em"
15+
preserveAspectRatio="xMidYMid meet"
16+
viewBox="0 0 24 24"
17+
>
18+
<path
19+
fill="currentColor"
20+
d="M5 21q-.825 0-1.412-.587T3 19V5q0-.825.588-1.412T5 3h5v2H5v14h14v-5.35l2 2V19q0 .825-.587 1.413T19 21H5Zm1-4l3-4l2.25 3l3-4L18 17H6Zm15.55-3.6l-3.1-3.1q-.525.35-1.125.525T16.05 11q-1.85 0-3.15-1.312T11.6 6.5q0-1.875 1.313-3.187T16.1 2q1.875 0 3.188 1.313T20.6 6.5q0 .675-.2 1.3t-.5 1.15L22.95 12l-1.4 1.4ZM16.1 9q1.05 0 1.775-.725T18.6 6.5q0-1.05-.725-1.775T16.1 4q-1.05 0-1.775.725T13.6 6.5q0 1.05.725 1.775T16.1 9Z"
21+
/>
22+
</svg>

packages/widgets/src/lib/components/PipelineIcon/PipelineIcon.svelte

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@
4141
import IconMaskGeneration from "../Icons/IconMaskGeneration.svelte";
4242
import IconTextTo3D from "../Icons/IconTextTo3D.svelte";
4343
import IconImageTo3D from "../Icons/IconImageTo3D.svelte";
44+
import IconImageFeatureExtraction from "../Icons/IconImageFeatureExtraction.svelte";
4445
import type { PipelineType } from "@huggingface/tasks";
4546
4647
export let classNames = "";
@@ -92,6 +93,7 @@
9293
"zero-shot-object-detection": IconZeroShotObjectDetection,
9394
"text-to-3d": IconTextTo3D,
9495
"image-to-3d": IconImageTo3D,
96+
"image-feature-extraction": IconImageFeatureExtraction,
9597
};
9698
9799
$: iconComponent =

0 commit comments

Comments
 (0)