Skip to content

Update recommended models for 2 tasks #883

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.

Sign up for GitHub

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 9 commits into from
Oct 16, 2024
Merged

Update recommended models for 2 tasks #883

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
240cb28
First draft
NielsRogge Aug 29, 2024
f392488
Update description
NielsRogge Sep 2, 2024
8679c09
Merge remote-tracking branch 'upstream/main' into update_tasks
NielsRogge Oct 8, 2024
6552456
Apply prettier
NielsRogge Oct 8, 2024
c4c130e
Update packages/tasks/src/tasks/object-detection/data.ts
NielsRogge Oct 16, 2024
a070b1b
Update packages/tasks/src/tasks/object-detection/data.ts
NielsRogge Oct 16, 2024
e1028f0
Update packages/tasks/src/tasks/image-segmentation/data.ts
NielsRogge Oct 16, 2024
b64f1dd
Update packages/tasks/src/tasks/image-segmentation/data.ts
NielsRogge Oct 16, 2024
876caee
Merge branch 'main' into update_tasks
NielsRogge Oct 16, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion packages/tasks/src/tasks/image-segmentation/about.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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",
});
```

Expand Down
13 changes: 7 additions & 6 deletions packages/tasks/src/tasks/image-segmentation/data.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down
9 changes: 5 additions & 4 deletions packages/tasks/src/tasks/object-detection/data.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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",
},
],
Expand Down
Loading