IMvision12 commited on
Commit
5aefffb
·
verified ·
1 Parent(s): a843ceb

docs: load image processor via from_weights

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -43,7 +43,7 @@ from PIL import Image
43
  from kerasformers.models.mask2former import Mask2FormerUniversalSegment, Mask2FormerImageProcessor
44
 
45
  model = Mask2FormerUniversalSegment.from_weights("kerasformers/mask2former-swin-tiny-coco-instance")
46
- processor = Mask2FormerImageProcessor()
47
 
48
  image = Image.open("your_image.jpg").convert("RGB")
49
  output = model(processor(image)["pixel_values"], training=False)
 
43
  from kerasformers.models.mask2former import Mask2FormerUniversalSegment, Mask2FormerImageProcessor
44
 
45
  model = Mask2FormerUniversalSegment.from_weights("kerasformers/mask2former-swin-tiny-coco-instance")
46
+ processor = Mask2FormerImageProcessor.from_weights("kerasformers/mask2former-swin-tiny-coco-instance")
47
 
48
  image = Image.open("your_image.jpg").convert("RGB")
49
  output = model(processor(image)["pixel_values"], training=False)