Update README.md

README.md

@@ -8,6 +8,20 @@ metrics:
 - mean_iou
 base_model:
 - nvidia/segformer-b1-finetuned-ade-512-512
+
+tags:
+  - scientific
+  - research
+citation: |
+  @article{author2023title,
+    title={Title of the Paper},
+    author={Author, A.},
+    journal={Journal Name},
+    year={2023},
+    url={https://doi.org/10.1234/example}
+  }
+source: https://doi.org/10.1234/example
+
 ---
 
 ## Usage
@@ -17,7 +31,7 @@ import torch, torch.nn.functional as F
 from PIL import Image
 import numpy as np
 
-repo = "GlobalWheat/GWFSS_model_v1.0"
+repo = "GlobalWheat/GWFSS_model_v1.1"
 processor = AutoImageProcessor.from_pretrained(repo)
 model = SegformerForSemanticSegmentation.from_pretrained(repo).eval()
 
@@ -29,4 +43,9 @@ with torch.no_grad():
 pred = up.argmax(1)[0].cpu().numpy()  # (H, W) class IDs
 ```
 
-This version is based on huggingface Segformer which could be slightly different from the one we used for our paper. The paper version was implemented based on the mmsegmentation. You can find the model weight for mmsegmentation library in this repo as well.
+This version is based on huggingface Segformer which could be slightly different from the one we used for our paper. The paper version was implemented based on the mmsegmentation. You can find the model weight for mmsegmentation library in this repo as well.
+
+## Related Paper
+This dataset is associated with the following paper:
+The Global Wheat Full Semantic Organ Segmentation (GWFSS) Dataset
+https://doi.org/10.1016/j.plaphe.2025.100084