Skip to content

Commit 72b0d9b

Browse files
rajaryan18pcuencacoyotte508
authored
📝 Update Mask Generation Task (#432)
Added about.md for Zero-shot-object-detection, text-to-3d and mask-generation. Also added use cases for them all. --------- Co-authored-by: Pedro Cuenca <[email protected]> Co-authored-by: Eliott C. <[email protected]> Co-authored-by: coyotte508 <[email protected]>
1 parent cd0e7c6 commit 72b0d9b

File tree

3 files changed

+28
-3
lines changed

3 files changed

+28
-3
lines changed

packages/tasks/.prettierignore

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
pnpm-lock.yaml
22
# In order to avoid code samples to have tabs, they don't display well on npm
33
README.md
4-
dist
4+
dist
5+
.tshy

packages/tasks/src/tasks/mask-generation/about.md

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,16 @@ Generating masks can facilitate learning, especially in semi or unsupervised lea
1212

1313
For applications where humans are in the loop, masks highlight certain regions of images for humans to validate.
1414

15+
### Medical Imaging
16+
17+
Mask generation models are used in medical imaging to aid in segmenting and analyzing specific regions.
18+
19+
### Autonomous Vehicles
20+
21+
Mask generation models are used to create segments and masks for obstacles and other objects in view.
22+
23+
This page was made possible thanks to the efforts of [Raj Aryan](https://huggingface.co/thatrajaryan) and other contributors.
24+
1525
## Task Variants
1626

1727
### Segmentation

packages/tasks/src/tasks/mask-generation/data.ts

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,16 @@
11
import type { TaskDataCustom } from "../index.js";
22

33
const taskData: TaskDataCustom = {
4-
datasets: [],
4+
datasets: [
5+
{
6+
description: "Widely used benchmark dataset for multiple Vision tasks.",
7+
id: "merve/coco2017",
8+
},
9+
{
10+
description: "Medical Imaging dataset of the Human Brain for segmentation and mask generating tasks",
11+
id: "rocky93/BraTS_segmentation",
12+
},
13+
],
514
demo: {
615
inputs: [
716
{
@@ -16,7 +25,12 @@ const taskData: TaskDataCustom = {
1625
},
1726
],
1827
},
19-
metrics: [],
28+
metrics: [
29+
{
30+
description: "IoU is used to measure the overlap between predicted mask and the ground truth mask.",
31+
id: "Intersection over Union (IoU)",
32+
},
33+
],
2034
models: [
2135
{
2236
description: "Small yet powerful mask generation model.",

0 commit comments

Comments
 (0)