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";
4848const inference = new HfInference (HF_TOKEN );
4949await inference .imageSegmentation ({
5050 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" ,
5252});
5353```
5454
Original file line number Diff line number Diff line change @@ -44,21 +44,22 @@ const taskData: TaskDataCustom = {
4444 models : [
4545 {
4646 // 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" ,
4950 } ,
5051 {
5152 description : "Background removal model." ,
5253 id : "briaai/RMBG-1.4" ,
5354 } ,
54- {
55- description : "Semantic segmentation model trained on ADE20k benchmark dataset with 512x512 resolution." ,
56- id : "nvidia/segformer-b0-finetuned-ade-512-512" ,
57- } ,
5855 {
5956 description : "A multipurpose image segmentation model for high resolution images." ,
6057 id : "ZhengPeng7/BiRefNet" ,
6158 } ,
59+ {
60+ description : "Semantic segmentation model trained on ADE20k dataset." ,
61+ id : "nvidia/segformer-b0-finetuned-ade-512-512" ,
62+ } ,
6263 {
6364 description : "Panoptic segmentation model trained COCO (common objects) dataset." ,
6465 id : "facebook/mask2former-swin-large-coco-panoptic" ,
Original file line number Diff line number Diff line change @@ -43,15 +43,16 @@ const taskData: TaskDataCustom = {
4343 ] ,
4444 models : [
4545 {
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 ." ,
4747 id : "facebook/detr-resnet-50" ,
4848 } ,
4949 {
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 " ,
5252 } ,
5353 {
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." ,
5556 id : "PekingU/rtdetr_r18vd_coco_o365" ,
5657 } ,
5758 ] ,
You can’t perform that action at this time.
0 commit comments