chmcbs commited on
Commit
ac9eaf5
·
verified ·
1 Parent(s): 3359f8d

Refine example usage

Browse files
Files changed (1) hide show
  1. README.md +2 -3
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.