zkep commited on
Commit
6cd1d87
·
1 Parent(s): 060ed34

Update README.md

Browse files

from transformers.image_transforms import rgb_to_id replace from transformers.models.detr.feature_extraction_detr import rgb_to_id

Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -48,7 +48,7 @@ import torch
48
  import numpy
49
 
50
  from transformers import DetrFeatureExtractor, DetrForSegmentation
51
- from transformers.models.detr.feature_extraction_detr import rgb_to_id
52
 
53
  url = "http://images.cocodataset.org/val2017/000000039769.jpg"
54
  image = Image.open(requests.get(url, stream=True).raw)
 
48
  import numpy
49
 
50
  from transformers import DetrFeatureExtractor, DetrForSegmentation
51
+ from transformers.image_transforms import rgb_to_id
52
 
53
  url = "http://images.cocodataset.org/val2017/000000039769.jpg"
54
  image = Image.open(requests.get(url, stream=True).raw)