Add files via upload

Author: senya-ashukha

colab/LaMa_inpainting.ipynb

@@ -17,6 +17,23 @@
     "accelerator": "GPU"
   },
   "cells": [
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "_pRpIwnaOnb3"
+      },
+      "source": [
+        "# 🦙 **LaMa: Resolution-robust Large Mask Inpainting with Fourier Convolutions**\n",
+        "\n",
+        "[[Project page](https://saic-mdal.github.io/lama-project/)] [[arXiv](https://arxiv.org/abs/2109.07161)] [[Supplementary](https://ashukha.com/projects/lama_21/lama_supmat_2021.pdf)] [[BibTeX](https://senya-ashukha.github.io/projects/lama_21/paper.txt)]\n",
+        "\n",
+        "<p align=\"center\" \"font-size:30px;\">\n",
+        "Our model generalizes surprisingly well to much higher resolutions (~2k❗️) than it saw during training (256x256), and achieves the excellent performance even in challenging scenarios, e.g. completion of periodic structures.\n",
+        "</p>\n",
+        "\n",
+        "# Try it yorself!👇\n"
+      ]
+    },
     {
       "cell_type": "code",
       "metadata": {
@@ -25,7 +42,7 @@
           "base_uri": "https://localhost:8080/"
         },
         "cellView": "form",
-        "outputId": "0c7de29c-0275-4d92-9407-fae5e3d232f5"
+        "outputId": "9570e776-baea-4157-acfb-a35ec1f00e92"
       },
       "source": [
         "#@title Run this sell to set everything up\n",
@@ -133,33 +150,57 @@
         "  with open(filename, 'wb') as f:\n",
         "    f.write(binary)"
       ],
-      "execution_count": null,
+      "execution_count": 1,
       "outputs": [
         {
           "output_type": "stream",
           "name": "stdout",
           "text": [
             "\n",
             "> Cloning the repo\n",
-            "fatal: destination path 'lama' already exists and is not an empty directory.\n",
+            "Cloning into 'lama'...\n",
+            "remote: Enumerating objects: 406, done.\u001b[K\n",
+            "remote: Counting objects: 100% (406/406), done.\u001b[K\n",
+            "remote: Compressing objects: 100% (341/341), done.\u001b[K\n",
+            "remote: Total 406 (delta 127), reused 261 (delta 45), pack-reused 0\u001b[K\n",
+            "Receiving objects: 100% (406/406), 11.76 MiB | 17.75 MiB/s, done.\n",
+            "Resolving deltas: 100% (127/127), done.\n",
             "\n",
             "> Install dependencies\n",
+            "\u001b[K     |████████████████████████████████| 12.5 MB 17 kB/s \n",
+            "\u001b[K     |████████████████████████████████| 22.3 MB 1.3 MB/s \n",
+            "\u001b[K     |████████████████████████████████| 72 kB 669 kB/s \n",
+            "\u001b[K     |████████████████████████████████| 144 kB 48.1 MB/s \n",
+            "\u001b[K     |████████████████████████████████| 841 kB 39.7 MB/s \n",
+            "\u001b[K     |████████████████████████████████| 271 kB 41.7 MB/s \n",
+            "\u001b[K     |████████████████████████████████| 45 kB 3.5 MB/s \n",
+            "\u001b[K     |████████████████████████████████| 37.1 MB 40 kB/s \n",
+            "\u001b[K     |████████████████████████████████| 948 kB 33.8 MB/s \n",
+            "\u001b[K     |████████████████████████████████| 112 kB 48.9 MB/s \n",
+            "\u001b[K     |████████████████████████████████| 74 kB 3.0 MB/s \n",
+            "\u001b[K     |████████████████████████████████| 176 kB 40.7 MB/s \n",
+            "\u001b[K     |████████████████████████████████| 829 kB 40.1 MB/s \n",
+            "\u001b[K     |████████████████████████████████| 125 kB 32.2 MB/s \n",
+            "\u001b[K     |████████████████████████████████| 269 kB 43.1 MB/s \n",
+            "\u001b[K     |████████████████████████████████| 1.3 MB 31.6 MB/s \n",
+            "\u001b[K     |████████████████████████████████| 160 kB 47.9 MB/s \n",
+            "\u001b[K     |████████████████████████████████| 271 kB 40.5 MB/s \n",
+            "\u001b[?25h  Building wheel for antlr4-python3-runtime (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
+            "  Building wheel for pyyaml (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
+            "  Building wheel for future (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
+            "  Building wheel for wget (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
             "\n",
             "> Changing the dir to:\n",
             "/content/lama\n",
             "\n",
             "> Download the model\n",
             "  % Total    % Received % Xferd  Average Speed   Time    Time     Time  Current\n",
             "                                 Dload  Upload   Total   Spent    Left  Speed\n",
-            "  0     0    0     0    0     0      0      0 --:--:-- --:--:-- --:--:--     0\n",
-            "100  340M    0  340M    0     0   9.7M      0 --:--:--  0:00:34 --:--:--  9.9M^C\n",
+            "  0     0    0     0    0     0      0      0 --:--:--  0:00:01 --:--:--     0\n",
+            "100  363M    0  363M    0     0  9852k      0 --:--:--  0:00:37 --:--:-- 9975k\n",
             "Archive:  big-lama.zip\n",
-            "  End-of-central-directory signature not found.  Either this file is not\n",
-            "  a zipfile, or it constitutes one disk of a multi-part archive.  In the\n",
-            "  latter case the central directory and zipfile comment will be found on\n",
-            "  the last disk(s) of this archive.\n",
-            "unzip:  cannot find zipfile directory in one of big-lama.zip or\n",
-            "        big-lama.zip.zip, and cannot find big-lama.zip.ZIP, period.\n",
+            "  inflating: big-lama/config.yaml    \n",
+            "  inflating: big-lama/models/best.ckpt  \n",
             "\n",
             "> Init mask-drawing code\n"
           ]