File tree Expand file tree Collapse file tree 3 files changed +13
-11
lines changed Expand file tree Collapse file tree 3 files changed +13
-11
lines changed Original file line number Diff line number Diff line change @@ -48,7 +48,7 @@ import { HfInference } from "@huggingface/inference";
48
48
const inference = new HfInference (HF_TOKEN );
49
49
await inference .imageSegmentation ({
50
50
data: await (await fetch (" https://picsum.photos/300/300" )).blob (),
51
- model: " facebook/detr-resnet-50 -panoptic" ,
51
+ model: " facebook/mask2former-swin-base-coco -panoptic" ,
52
52
});
53
53
```
54
54
Original file line number Diff line number Diff line change @@ -44,21 +44,22 @@ const taskData: TaskDataCustom = {
44
44
models : [
45
45
{
46
46
// TO DO: write description
47
- description : "Solid panoptic segmentation model trained on the COCO 2017 benchmark dataset." ,
48
- id : "facebook/detr-resnet-50-panoptic" ,
47
+ description :
48
+ "Solid semantic segmentation model trained on ADE20k." ,
49
+ id : "openmmlab/upernet-convnext-small" ,
49
50
} ,
50
51
{
51
52
description : "Background removal model." ,
52
53
id : "briaai/RMBG-1.4" ,
53
54
} ,
54
- {
55
- description : "Semantic segmentation model trained on ADE20k benchmark dataset with 512x512 resolution." ,
56
- id : "nvidia/segformer-b0-finetuned-ade-512-512" ,
57
- } ,
58
55
{
59
56
description : "A multipurpose image segmentation model for high resolution images." ,
60
57
id : "ZhengPeng7/BiRefNet" ,
61
58
} ,
59
+ {
60
+ description : "Semantic segmentation model trained on ADE20k dataset." ,
61
+ id : "nvidia/segformer-b0-finetuned-ade-512-512" ,
62
+ } ,
62
63
{
63
64
description : "Panoptic segmentation model trained COCO (common objects) dataset." ,
64
65
id : "facebook/mask2former-swin-large-coco-panoptic" ,
Original file line number Diff line number Diff line change @@ -43,15 +43,16 @@ const taskData: TaskDataCustom = {
43
43
] ,
44
44
models : [
45
45
{
46
- description : "Solid object detection model trained on the benchmark dataset COCO 2017." ,
46
+ description : "Solid object detection model pre- trained on the COCO 2017 dataset ." ,
47
47
id : "facebook/detr-resnet-50" ,
48
48
} ,
49
49
{
50
- description : "Strong object detection model trained on ImageNet-21k dataset ." ,
51
- id : "microsoft/beit-base-patch16-224-pt22k-ft22k " ,
50
+ description : "Real-time and accurate object detection model." ,
51
+ id : "jameslahm/yolov10x " ,
52
52
} ,
53
53
{
54
- description : "Fast and accurate object detection model trained on COCO dataset." ,
54
+ description :
55
+ "Fast and accurate object detection model trained on COCO and Object365 datasets." ,
55
56
id : "PekingU/rtdetr_r18vd_coco_o365" ,
56
57
} ,
57
58
] ,
You can’t perform that action at this time.
0 commit comments