Refine example usage
Browse files
README.md
CHANGED
|
@@ -60,7 +60,7 @@ predictions = model.predict(image)
|
|
| 60 |
|
| 61 |
# Count heads
|
| 62 |
num_heads = model.count_heads(predictions)
|
| 63 |
-
print(f"🌾 {num_heads} heads detected
|
| 64 |
|
| 65 |
# Create visualisation
|
| 66 |
overlay = model.overlay_mask(image, predictions, alpha=0.5, heads_only=True)
|
|
@@ -76,5 +76,4 @@ Best performance is achieved with overhead imagery under diffuse lighting. Known
|
|
| 76 |
|
| 77 |
## Training Data
|
| 78 |
|
| 79 |
-
This model was trained on [`GWFSS_v1.0_labelled`](https://huggingface.co/datasets/GlobalWheat/GWFSS_v1.0) from the
|
| 80 |
-
[Global Wheat Full Semantic Organ Segmentation](https://doi.org/10.1016/j.plaphe.2025.100084) dataset.
|
|
|
|
| 60 |
|
| 61 |
# Count heads
|
| 62 |
num_heads = model.count_heads(predictions)
|
| 63 |
+
print(f"🌾 {num_heads} heads detected")
|
| 64 |
|
| 65 |
# Create visualisation
|
| 66 |
overlay = model.overlay_mask(image, predictions, alpha=0.5, heads_only=True)
|
|
|
|
| 76 |
|
| 77 |
## Training Data
|
| 78 |
|
| 79 |
+
This model was trained on [`GWFSS_v1.0_labelled`](https://huggingface.co/datasets/GlobalWheat/GWFSS_v1.0) from the [Global Wheat Full Semantic Organ Segmentation](https://doi.org/10.1016/j.plaphe.2025.100084) dataset.
|
|
|