Added visual aids for architecture

Rishit-dagli · Dec 20, 2020 · eb5bde7 · eb5bde7
1 parent 5aa57cb
commit eb5bde7
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/MIRNet_Saved_Model.ipynb b/MIRNet_Saved_Model.ipynb
@@ -6,7 +6,7 @@
       "name": "MIRNet_Saved_Model.ipynb",
       "provenance": [],
       "collapsed_sections": [],
-      "authorship_tag": "ABX9TyOQaWgNpJEfqSNyRLoKGmEo",
+      "authorship_tag": "ABX9TyOutMPPkom13uQX+4gm3WiK",
       "include_colab_link": true
     },
     "kernelspec": {
@@ -47,12 +47,19 @@
         "- spatial and channel attention mechanisms for capturing contextual information\r\n",
         "- attention based multi-scale feature aggregation.\r\n",
         "\r\n",
+        "![](https://i.imgur.com/VOzfG9a.png)\r\n",
+        "<p align=\"center\">\r\n",
+        "<small>Framework of MIRNet</small>\r\n",
+        "</p>\r\n",
+        "\r\n",
+        "The figure above shows the framework of the proposed network MIRNet that learns enriched feature representations for image restoration and enhancement.\r\n",
+        "\r\n",
         "<p align=\"center\">\r\n",
         "<img src=\"https://github.com/soumik12345/MIRNet/raw/master/assets/lol_results.gif\"</img><br>\r\n",
         "<small><a href=\"https://github.com/soumik12345/MIRNet/blob/master/assets/lol_results.gif\">Source</a></small>\r\n",
         "</p>\r\n",
         "\r\n",
-        "Model training code and pre-trained weights are provided by Soumik through [this repository](https://github.com/soumik12345/MIRNet/). "
+        "Model training code and pre-trained weights are provided by Soumik through [this repository](https://github.com/soumik12345/MIRNet/)."
       ]
     },
     {