Update README.md

README.md

@@ -7,4 +7,74 @@ metrics:
 pipeline_tag: image-classification
 tags:
 - climate
----
+---
+
+## Model description
+
+This is a transformers based image classification model, implemented using the technique of transfer learning.
+The pretrained model is [Vision transformer](https://huggingface.co/google/vit-base-patch16-224) trained on Imagenet-21k.
+
+## Datasets
+
+The dataset used is downloaded from git repo [Agri-Hub/Space2Ground](https://github.com/Agri-Hub/Space2Ground/tree/main).
+I used Street-level image patches folder for this model. It is a dataset containing cropped vegetation parts of 
+mapillary street-level images. Further details are on the linked git repo.
+
+### How to use
+
+You can use this model directly with help of pipeline class from transformers library of hugging face
+
+```python
+
+>>>from transformers import pipeline
+>>>classifier = pipeline("image-classification", model="iammartian0/vegetation_classification_model")
+>>>classifier(image)
+
+```
+
+## Training procedure
+
+
+
+### Preprocessing
+
+ Assigining labels based on parent folder names
+
+### Image Transformations
+
+Applied RandomResizedCrop from torchvision.transforms to all the training images.
+
+### Finetuning
+
+Model is finetuned on the dataset for four epochs
+
+## Evaluation results
+
+
+
+
+### BibTeX entry and citation info
+
+```bibtex
+@article{DBLP:journals/corr/abs-1810-04805,
+  author    = {Jacob Devlin and
+               Ming{-}Wei Chang and
+               Kenton Lee and
+               Kristina Toutanova},
+  title     = {{BERT:} Pre-training of Deep Bidirectional Transformers for Language
+               Understanding},
+  journal   = {CoRR},
+  volume    = {abs/1810.04805},
+  year      = {2018},
+  url       = {http://arxiv.org/abs/1810.04805},
+  archivePrefix = {arXiv},
+  eprint    = {1810.04805},
+  timestamp = {Tue, 30 Oct 2018 20:39:56 +0100},
+  biburl    = {https://dblp.org/rec/journals/corr/abs-1810-04805.bib},
+  bibsource = {dblp computer science bibliography, https://dblp.org}
+}
+```
+
+<a href="https://huggingface.co/exbert/?model=bert-base-cased">
+	<img width="300px" src="https://cdn-media.huggingface.co/exbert/button.png">
+</a>