diff --git a/packages/tasks/src/tasks/image-segmentation/about.md b/packages/tasks/src/tasks/image-segmentation/about.md index 4a8a45a19..ce9be3aec 100644 --- a/packages/tasks/src/tasks/image-segmentation/about.md +++ b/packages/tasks/src/tasks/image-segmentation/about.md @@ -48,7 +48,7 @@ import { HfInference } from "@huggingface/inference"; const inference = new HfInference(HF_TOKEN); await inference.imageSegmentation({ data: await (await fetch("https://picsum.photos/300/300")).blob(), - model: "facebook/detr-resnet-50-panoptic", + model: "facebook/mask2former-swin-base-coco-panoptic", }); ``` diff --git a/packages/tasks/src/tasks/image-segmentation/data.ts b/packages/tasks/src/tasks/image-segmentation/data.ts index 8648eaff0..17c5b7d0a 100644 --- a/packages/tasks/src/tasks/image-segmentation/data.ts +++ b/packages/tasks/src/tasks/image-segmentation/data.ts @@ -44,21 +44,22 @@ const taskData: TaskDataCustom = { models: [ { // TO DO: write description - description: "Solid panoptic segmentation model trained on the COCO 2017 benchmark dataset.", - id: "facebook/detr-resnet-50-panoptic", + description: + "Solid semantic segmentation model trained on ADE20k.", + id: "openmmlab/upernet-convnext-small", }, { description: "Background removal model.", id: "briaai/RMBG-1.4", }, - { - description: "Semantic segmentation model trained on ADE20k benchmark dataset with 512x512 resolution.", - id: "nvidia/segformer-b0-finetuned-ade-512-512", - }, { description: "A multipurpose image segmentation model for high resolution images.", id: "ZhengPeng7/BiRefNet", }, + { + description: "Semantic segmentation model trained on ADE20k dataset.", + id: "nvidia/segformer-b0-finetuned-ade-512-512", + }, { description: "Panoptic segmentation model trained COCO (common objects) dataset.", id: "facebook/mask2former-swin-large-coco-panoptic", diff --git a/packages/tasks/src/tasks/object-detection/data.ts b/packages/tasks/src/tasks/object-detection/data.ts index 10a25cda7..80f87a590 100644 --- a/packages/tasks/src/tasks/object-detection/data.ts +++ b/packages/tasks/src/tasks/object-detection/data.ts @@ -43,15 +43,16 @@ const taskData: TaskDataCustom = { ], models: [ { - description: "Solid object detection model trained on the benchmark dataset COCO 2017.", + description: "Solid object detection model pre-trained on the COCO 2017 dataset.", id: "facebook/detr-resnet-50", }, { - description: "Strong object detection model trained on ImageNet-21k dataset.", - id: "microsoft/beit-base-patch16-224-pt22k-ft22k", + description: "Real-time and accurate object detection model.", + id: "jameslahm/yolov10x", }, { - description: "Fast and accurate object detection model trained on COCO dataset.", + description: + "Fast and accurate object detection model trained on COCO and Object365 datasets.", id: "PekingU/rtdetr_r18vd_coco_o365", }, ],