Update README.md

README.md

@@ -8,6 +8,14 @@ base_model:
 
 # Research paper: Segmentation of Wheat Rust Disease Using Co-Salient Feature Extraction
 can be accessed via https://doi.org/10.3390/agriengineering7020023 
+
+It is based on a two-stage model:
+--ViT based classification and DCFM based co-salient feature extraction for segmenting out wheat rust disease.
+
+#ViT
+Vision Transformer. 
+
+
 # DCFM
 The official repo of the paper `Democracy Does Matter: Comprehensive Feature Mining for Co-Salient Object Detection`.