IMvision12 commited on
Commit
ff876b3
verified
1 Parent(s): 9f0f37d

fix readme.md

Browse files
Files changed (1) hide show
  1. README.md +5 -4
README.md CHANGED
@@ -25,7 +25,8 @@ tags:
25
 
26
  Paper: [Per-Pixel Classification is Not All You Need for Semantic Segmentation (arXiv:2107.06278)](https://arxiv.org/abs/2107.06278) 路 [HF Papers](https://huggingface.co/papers/2107.06278)
27
 
28
- MaskFormer reframes segmentation as mask classification: a backbone and pixel decoder feed a transformer decoder whose queries each predict a binary mask and a class. One architecture covers semantic, instance, and panoptic outputs via post-processing.\n
 
29
  For more details on the model, please go to the upstream [model card](https://huggingface.co/facebook/maskformer-swin-tiny-coco).
30
 
31
  Pure-**Keras 3** conversion of [`facebook/maskformer-swin-tiny-coco`](https://huggingface.co/facebook/maskformer-swin-tiny-coco) for [kerasformers](https://github.com/IMvision12/KerasFormers). One implementation runs unmodified on **TensorFlow / Torch / JAX**.
@@ -39,10 +40,10 @@ import os
39
  os.environ["KERAS_BACKEND"] = "torch" # or "jax" / "tensorflow"
40
 
41
  from PIL import Image
42
- from {meta['import_path']} import {meta['load_cls']}, {meta['proc_cls']}
43
 
44
- model = {meta["load_cls"]}.from_weights("kerasformers/{variant}")
45
- processor = {meta["proc_cls"]}.from_weights("kerasformers/{variant}")
46
 
47
  image = Image.open("your_image.jpg").convert("RGB")
48
  output = model(processor(image)["pixel_values"], training=False)
 
25
 
26
  Paper: [Per-Pixel Classification is Not All You Need for Semantic Segmentation (arXiv:2107.06278)](https://arxiv.org/abs/2107.06278) 路 [HF Papers](https://huggingface.co/papers/2107.06278)
27
 
28
+ MaskFormer reframes segmentation as mask classification: a backbone and pixel decoder feed a transformer decoder whose queries each predict a binary mask and a class. One architecture covers semantic, instance, and panoptic outputs via post-processing.
29
+
30
  For more details on the model, please go to the upstream [model card](https://huggingface.co/facebook/maskformer-swin-tiny-coco).
31
 
32
  Pure-**Keras 3** conversion of [`facebook/maskformer-swin-tiny-coco`](https://huggingface.co/facebook/maskformer-swin-tiny-coco) for [kerasformers](https://github.com/IMvision12/KerasFormers). One implementation runs unmodified on **TensorFlow / Torch / JAX**.
 
40
  os.environ["KERAS_BACKEND"] = "torch" # or "jax" / "tensorflow"
41
 
42
  from PIL import Image
43
+ from kerasformers.models.maskformer import MaskFormerUniversalSegment, MaskFormerImageProcessor
44
 
45
+ model = MaskFormerUniversalSegment.from_weights("kerasformers/maskformer-swin-tiny-coco")
46
+ processor = MaskFormerImageProcessor.from_weights("kerasformers/maskformer-swin-tiny-coco")
47
 
48
  image = Image.open("your_image.jpg").convert("RGB")
49
  output = model(processor(image)["pixel_values"], training=False)