Update README.md

README.md

@@ -41,17 +41,47 @@ You can use the following tags to further select images for testing:
 "coronal holes", "loops" or "flares"
 You can also choose "active regions" to get a general pool for testing.
 
-#### NASA_SDO_Coronal_Hole
+### NASA_SDO_Coronal_Hole
 
 ![NASA_SDO_Coronal_Hole](images/NASA_SDO_Coronal_Hole2.jpg)
 
-#### NASA_SDO_Coronal_Loop
+### NASA_SDO_Coronal_Loop
 
 ![NASA_SDO_Coronal_Loop](images/NASA_SDO_Coronal_Loop.jpg)
 
-#### NASA_SDO_Solar_Flare
+### NASA_SDO_Solar_Flare
 
 ![NASA_SDO_Solar_Flare](images/NASA_SDO_Solar_Flare.jpg)
 
+## Training data
+The ViT model was pretrained on a dataset consisting of 14 million images and 21k classes ([ImageNet-21k](http://www.image-net.org/).
+More information on the base model used can be found here: (https://huggingface.co/google/vit-base-patch16-224-in21k)
+
 ## How to use this Model
 
+## References
+
+A publication on this work is currently in preparation. In the meantime, please refer to this model by using the following citation: 
+```
+@misc{sdovt2022,
+  author = {Frank Soboczenski and Paul J Wright},
+  title = {SDOVT: A Vision Transformer Model for Solar Dynamics Observatory (SDO) Data},
+  url = {http://github.com/h21k/sdovis},
+  version = {1.0},
+  year = {2022},
+}
+```
+
+For the base ViT model used please refer to:
+
+```bibtex
+@misc{wu2020visual,
+      title={Visual Transformers: Token-based Image Representation and Processing for Computer Vision}, 
+      author={Bichen Wu and Chenfeng Xu and Xiaoliang Dai and Alvin Wan and Peizhao Zhang and Zhicheng Yan and Masayoshi Tomizuka and Joseph Gonzalez and Kurt Keutzer and Peter Vajda},
+      year={2020},
+      eprint={2006.03677},
+      archivePrefix={arXiv},
+      primaryClass={cs.CV}
+}
+```
+