Update README.md with new model card content
Browse files
README.md
CHANGED
|
@@ -9,8 +9,8 @@ Weights are released under the [MIT License](https://opensource.org/license/mit)
|
|
| 9 |
|
| 10 |
## Links
|
| 11 |
|
| 12 |
-
* [
|
| 13 |
-
* [SegFormer API Documentation](
|
| 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 |
-
|
|
| 35 |
-
|
|
| 36 |
-
|
|
| 37 |
-
|
|
| 38 |
-
|
|
| 39 |
-
|
|
| 40 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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
|