Update README.md with new model card content
Browse files
README.md
CHANGED
@@ -9,18 +9,19 @@ Weights are released under the [MIT License](https://opensource.org/license/mit)
|
|
9 |
|
10 |
## Links
|
11 |
|
12 |
-
* [Segmentation Quickstart Notebook](
|
13 |
-
* [SegFormer API Documentation](
|
14 |
* [SegFormer Model Card](https://huggingface.co/docs/transformers/en/model_doc/segformer)
|
15 |
-
|
|
|
16 |
|
17 |
## Installation
|
18 |
|
19 |
-
Keras and
|
20 |
|
21 |
```
|
22 |
-
pip install -U -q keras-
|
23 |
-
pip install -U -q keras
|
24 |
```
|
25 |
|
26 |
Jax, TensorFlow, and Torch come preinstalled in Kaggle Notebooks. For instructions on installing them in another environment see the [Keras Getting Started](https://keras.io/getting_started/) page.
|
@@ -38,22 +39,6 @@ The following model checkpoints are provided by the Keras team. Full code exampl
|
|
38 |
| segformer_b4 | 61.37M | SegFormer model with MiTB4 backbone. |
|
39 |
| segformer_b5 | 81.97M | SegFormer model with MiTB5 backbone. |
|
40 |
|
41 |
-
## Example code
|
42 |
-
|
43 |
-
```Python
|
44 |
-
import keras_cv
|
45 |
-
|
46 |
-
images = np.ones(shape=(1, 224, 224, 3))
|
47 |
-
labels = np.zeros(shape=(1, 224, 224, 1))
|
48 |
-
|
49 |
-
|
50 |
-
model = keras_cv.models.SegFormer.from_preset(
|
51 |
-
"segformer_b0", num_classes=2
|
52 |
-
)
|
53 |
-
# Evaluate model
|
54 |
-
model(images)
|
55 |
-
```
|
56 |
-
|
57 |
## Example Usage
|
58 |
```python
|
59 |
import keras_cv
|
|
|
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/)
|
17 |
|
18 |
## Installation
|
19 |
|
20 |
+
Keras and KerasHub can be installed with:
|
21 |
|
22 |
```
|
23 |
+
pip install -U -q keras-hub
|
24 |
+
pip install -U -q keras
|
25 |
```
|
26 |
|
27 |
Jax, TensorFlow, and Torch come preinstalled in Kaggle Notebooks. For instructions on installing them in another environment see the [Keras Getting Started](https://keras.io/getting_started/) page.
|
|
|
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
|
44 |
import keras_cv
|