Skip to content

📝 Update Mask Generation Task #432

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 11 commits into from
Nov 15, 2024
3 changes: 2 additions & 1 deletion packages/tasks/.prettierignore
Original file line number Diff line number Diff line change
@@ -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
dist
.tshy
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