diff --git a/packages/tasks/.prettierignore b/packages/tasks/.prettierignore index cac0c69496..09c2c88825 100644 --- a/packages/tasks/.prettierignore +++ b/packages/tasks/.prettierignore @@ -1,4 +1,5 @@ pnpm-lock.yaml # In order to avoid code samples to have tabs, they don't display well on npm README.md -dist \ No newline at end of file +dist +.tshy \ No newline at end of file diff --git a/packages/tasks/src/tasks/mask-generation/about.md b/packages/tasks/src/tasks/mask-generation/about.md index 18d6d38db4..2fa3a65d22 100644 --- a/packages/tasks/src/tasks/mask-generation/about.md +++ b/packages/tasks/src/tasks/mask-generation/about.md @@ -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 diff --git a/packages/tasks/src/tasks/mask-generation/data.ts b/packages/tasks/src/tasks/mask-generation/data.ts index 133e9a7f69..cc18380dbe 100644 --- a/packages/tasks/src/tasks/mask-generation/data.ts +++ b/packages/tasks/src/tasks/mask-generation/data.ts @@ -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: [ { @@ -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.",