Skip to content
10 changes: 10 additions & 0 deletions packages/tasks/src/tasks/mask-generation/about.md
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,16 @@ Generating masks can facilitate learning, especially in semi or unsupervised lea

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

### Medical Imaging

Mask generation models are used in medical imaging to aid in segmenting and analyzing specific regions.

### Autonomous Vehicles

Mask generation models are used to create segments and masks for obstacles and other objects in view.

This page was made possible thanks to the efforts of [Raj Aryan](https://huggingface.co/thatrajaryan) and other contributors.

## Task Variants

### Segmentation
Expand Down
18 changes: 16 additions & 2 deletions packages/tasks/src/tasks/mask-generation/data.ts
Original file line number Diff line number Diff line change
@@ -1,7 +1,16 @@
import type { TaskDataCustom } from "../index.js";

const taskData: TaskDataCustom = {
datasets: [],
datasets: [
{
description: "Widely used benchmark dataset for multiple Vision tasks.",
id: "merve/coco2017",
},
{
description: "Medical Imaging dataset of the Human Brain for segmentation and mask generating tasks",
id: "rocky93/BraTS_segmentation"
},
],
demo: {
inputs: [
{
Expand All @@ -16,7 +25,12 @@ const taskData: TaskDataCustom = {
},
],
},
metrics: [],
metrics: [
{
description: "IoU is used to measure the overlap between predicted mask and the ground truth mask.",
id: "Intersection over Union (IoU)",
},
],
models: [
{
description: "Small yet powerful mask generation model.",
Expand Down
Loading