facebook
/

detr-resnet-50-panoptic

Image Segmentation

Inference Endpoints

Model card Files Files and versions Community

Update README.md

#1

by zkep - opened Feb 16, 2023

base: refs/heads/main

←

from: refs/pr/1

Discussion Files changed

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -48,7 +48,7 @@ import torch
 import numpy
 from transformers import DetrFeatureExtractor, DetrForSegmentation
-from transformers.models.detr.feature_extraction_detr import rgb_to_id
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)

 import numpy
 from transformers import DetrFeatureExtractor, DetrForSegmentation
+from transformers.image_transforms import rgb_to_id
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)