Fixing README file

README.md

@@ -2,6 +2,20 @@
 
 This is the official code of the ICRL 2021 paper *Intrinsic-Extrinsic Convolution and Pooling for Learning on 3D Protein Structures*.
 
+![teaser](https://github.com/phermosilla/IEConv_proteins/blob/master/imgs/conv.png)
+![teaser](https://github.com/phermosilla/IEConv_proteins/blob/master/imgs/pooling.png)
+
+### Citation
+
+If you find this code useful please consider citing us:
+
+        @article{hermosilla2021ieconv,
+          title={Intrinsic-Extrinsic Convolution and Pooling for Learning on 3D Protein Structures},
+          author={Hermosilla, Pedro and Schäfer, Marco and Lang, Matěj and Fackelmann, Gloria and Vázquez, Pere Pau and Kozlíková, Barbora and Krone, Michael and Ritschel, Tobias and Ropinski, Timo},
+          journal={International Conference on Learning Representations},
+          year={2021}
+        }
+
 ### Instalation
 
 Open a docker container with the following command:
@@ -82,14 +96,3 @@ To evaluate the trained model execute:
     python Test.py --configFile confs/test.ini
 
 ### Trained models comming soon
-
-### Citation
-
-If you find this code useful please consider citing us:
-
-        @article{hermosilla2021ieconv,
-          title={Intrinsic-Extrinsic Convolution and Pooling for Learning on 3D Protein Structures},
-          author={Hermosilla, Pedro and Schäfer, Marco and Lang, Matěj and Fackelmann, Gloria and Vázquez, Pere Pau and Kozlíková, Barbora and Krone, Michael and Ritschel, Tobias and Ropinski, Timo},
-          journal={International Conference on Learning Representations},
-          year={2021}
-        }