Divyasreepat commited on
Commit
79d38a4
·
verified ·
1 Parent(s): b0c6744

Update README.md with new model card content

Browse files
Files changed (1) hide show
  1. README.md +14 -9
README.md CHANGED
@@ -9,8 +9,8 @@ Weights are released under the [MIT License](https://opensource.org/license/mit)
9
 
10
  ## Links
11
 
12
- * [Segmentation Quickstart Notebook](coming soon)
13
- * [SegFormer API Documentation](coming soon)
14
  * [SegFormer Model Card](https://huggingface.co/docs/transformers/en/model_doc/segformer)
15
  * [KerasHub Beginner Guide](https://keras.io/guides/keras_hub/getting_started/)
16
  * [KerasHub Model Publishing Guide](https://keras.io/guides/keras_hub/upload/)
@@ -31,13 +31,18 @@ Jax, TensorFlow, and Torch come preinstalled in Kaggle Notebooks. For instructio
31
  The following model checkpoints are provided by the Keras team. Full code examples for each are available below.
32
  | Preset name | Parameters | Description |
33
  |-----------------------|------------|---------------------------------------------------|
34
- | segformer_b0_imagenet | 3.72M | SegFormer model with a pretrained MiTB0 backbone. |
35
- | segformer_b0 | 3.72M | SegFormer model with MiTB0 backbone. |
36
- | segformer_b1 | 13.68M | SegFormer model with MiTB1 backbone. |
37
- | segformer_b2 | 24.73M | SegFormer model with MiTB2 backbone. |
38
- | segformer_b3 | 44.60M | SegFormer model with MiTB3 backbone. |
39
- | segformer_b4 | 61.37M | SegFormer model with MiTB4 backbone. |
40
- | segformer_b5 | 81.97M | SegFormer model with MiTB5 backbone. |
 
 
 
 
 
41
 
42
  ## Example Usage
43
  ```python
 
9
 
10
  ## Links
11
 
12
+ * [SegFormer Quickstart Notebook](https://www.kaggle.com/code/laxmareddypatlolla/segformer-quickstart-notebook)
13
+ * [SegFormer API Documentation](https://keras.io/keras_hub/api/models/segformer/)
14
  * [SegFormer Model Card](https://huggingface.co/docs/transformers/en/model_doc/segformer)
15
  * [KerasHub Beginner Guide](https://keras.io/guides/keras_hub/getting_started/)
16
  * [KerasHub Model Publishing Guide](https://keras.io/guides/keras_hub/upload/)
 
31
  The following model checkpoints are provided by the Keras team. Full code examples for each are available below.
32
  | Preset name | Parameters | Description |
33
  |-----------------------|------------|---------------------------------------------------|
34
+ | segformer_b0_ade20k_512 | 3.72M | SegFormer model with MiTB0 backbone fine-tuned on ADE20k in 512x512 resolution. |
35
+ | segformer_b1_ade20k_512 | 13.68M | SegFormer model with MiTB0 backbone fine-tuned on ADE20k in 512x512 resolution. |
36
+ | segformer_b2_ade20k_512 | 24.73M | SegFormer model with MiTB2 backbone fine-tuned on ADE20k in 512x512 resolution. |
37
+ | segformer_b3_ade20k_512 | 44.60M | SegFormer model with MiTB2 backbone fine-tuned on ADE20k in 512x512 resolution. |
38
+ | segformer_b4_ade20k_512 | 61.37M |SegFormer model with MiTB2 backbone fine-tuned on ADE20k in 512x512 resolution. |
39
+ | segformer_b5_ade20k_640 | 81.97M | SegFormer model with MiTB5 backbone fine-tuned on ADE20k in 640x640 resolution. |
40
+ | segformer_b0_cityscapes_1024 | 3.72M | SegFormer model with MiTB0 backbone fine-tuned on Cityscapes in 1024x1024 resolution. |
41
+ | segformer_b1_cityscapes_1024 | 13.68M | SegFormer model with MiTB0 backbone fine-tuned on Cityscapes in 1024x1024 resolution. |
42
+ | segformer_b2_cityscapes_1024 | 24.73M | SegFormer model with MiTB0 backbone fine-tuned on Cityscapes in 1024x1024 resolution. |
43
+ | segformer_b3_cityscapes_1024 | 44.60M | SegFormer model with MiTB0 backbone fine-tuned on Cityscapes in 1024x1024 resolution. |
44
+ | segformer_b4_cityscapes_1024 | 61.37M | SegFormer model with MiTB0 backbone fine-tuned on Cityscapes in 1024x1024 resolution. |
45
+ | segformer_b5_cityscapes_1024 | 81.97M | SegFormer model with MiTB0 backbone fine-tuned on Cityscapes in 1024x1024 resolution. |
46
 
47
  ## Example Usage
48
  ```python