Run-GFPGAN.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "source": [
    "# Setup\n",
    "\n",
    "These guidelines were made using the guidelines found on the original github for GFPGAN \n",
    "> [[Paper](https://arxiv.org/abs/2101.04061)] &emsp; [[Project Page](https://xinntao.github.io/projects/gfpgan)] &emsp; Authors:\n",
    "> [Xintao Wang](https://xinntao.github.io/), [Yu Li](https://yu-li.github.io/), [Honglun Zhang](https://scholar.google.com/citations?hl=en&user=KjQLROoAAAAJ), [Ying Shan](https://scholar.google.com/citations?user=4oXBp9UAAAAJ&hl=en) <br>\n",
    "> Applied Research Center (ARC), Tencent PCG\n",
    "\n",
    "First, install the required libraries"
   ],
   "metadata": {
    "gradient": {
     "editing": false,
     "id": "e12571bf-9d79-4003-aed3-39298bb91356",
     "kernelId": "a5a7e424-6201-4e2d-8ab7-cb22b6774ff4"
    }
   }
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "source": [
    "# Install basicsr - https://github.com/xinntao/BasicSR\n",
    "# We use BasicSR for both training and inference\n",
    "!pip install basicsr==1.3.5\n",
    "\n",
    "# Install facexlib - https://github.com/xinntao/facexlib\n",
    "# We use face detection and face restoration helper in the facexlib package\n",
    "!pip install facexlib\n",
    "\n",
    "\n",
    "# If you want to enhance the background (non-face) regions with Real-ESRGAN,\n",
    "# you also need to install the realesrgan package,\n",
    "# you must install this version to use realesrgan\n",
    "!pip install realesrgan==0.2.5"
   ],
   "outputs": [
    {
     "output_type": "stream",
     "name": "stdout",
     "text": [
      "Collecting basicsr==1.3.5\n",
      "  Downloading basicsr-1.3.5.tar.gz (161 kB)\n",
      "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m161.4/161.4 kB\u001b[0m \u001b[31m21.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25h  Preparing metadata (setup.py) ... \u001b[?25ldone\n",
      "\u001b[?25hRequirement already satisfied: addict in /usr/local/lib/python3.9/dist-packages (from basicsr==1.3.5) (2.4.0)\n",
      "Requirement already satisfied: future in /usr/lib/python3/dist-packages (from basicsr==1.3.5) (0.18.2)\n",
      "Requirement already satisfied: lmdb in /usr/local/lib/python3.9/dist-packages (from basicsr==1.3.5) (1.3.0)\n",
      "Requirement already satisfied: numpy in /usr/local/lib/python3.9/dist-packages (from basicsr==1.3.5) (1.20.3)\n",
      "Requirement already satisfied: opencv-python in /usr/local/lib/python3.9/dist-packages (from basicsr==1.3.5) (4.5.5.64)\n",
      "Requirement already satisfied: Pillow in /usr/local/lib/python3.9/dist-packages (from basicsr==1.3.5) (9.2.0)\n",
      "Requirement already satisfied: pyyaml in /usr/local/lib/python3.9/dist-packages (from basicsr==1.3.5) (5.4.1)\n",
      "Requirement already satisfied: requests in /usr/local/lib/python3.9/dist-packages (from basicsr==1.3.5) (2.28.1)\n",
      "Requirement already satisfied: scikit-image in /usr/local/lib/python3.9/dist-packages (from basicsr==1.3.5) (0.19.3)\n",
      "Requirement already satisfied: scipy in /usr/local/lib/python3.9/dist-packages (from basicsr==1.3.5) (1.8.1)\n",
      "Requirement already satisfied: tb-nightly in /usr/local/lib/python3.9/dist-packages (from basicsr==1.3.5) (2.11.0a20220824)\n",
      "Requirement already satisfied: torch>=1.7 in /usr/local/lib/python3.9/dist-packages (from basicsr==1.3.5) (1.12.0+cu116)\n",
      "Requirement already satisfied: torchvision in /usr/local/lib/python3.9/dist-packages (from basicsr==1.3.5) (0.13.0+cu116)\n",
      "Requirement already satisfied: tqdm in /usr/local/lib/python3.9/dist-packages (from basicsr==1.3.5) (4.64.0)\n",
      "Requirement already satisfied: yapf in /usr/local/lib/python3.9/dist-packages (from basicsr==1.3.5) (0.32.0)\n",
      "Requirement already satisfied: typing-extensions in /usr/local/lib/python3.9/dist-packages (from torch>=1.7->basicsr==1.3.5) (4.3.0)\n",
      "Requirement already satisfied: idna<4,>=2.5 in /usr/lib/python3/dist-packages (from requests->basicsr==1.3.5) (2.8)\n",
      "Requirement already satisfied: urllib3<1.27,>=1.21.1 in /usr/local/lib/python3.9/dist-packages (from requests->basicsr==1.3.5) (1.26.10)\n",
      "Requirement already satisfied: certifi>=2017.4.17 in /usr/lib/python3/dist-packages (from requests->basicsr==1.3.5) (2019.11.28)\n",
      "Requirement already satisfied: charset-normalizer<3,>=2 in /usr/local/lib/python3.9/dist-packages (from requests->basicsr==1.3.5) (2.1.0)\n",
      "Requirement already satisfied: networkx>=2.2 in /usr/local/lib/python3.9/dist-packages (from scikit-image->basicsr==1.3.5) (2.8.4)\n",
      "Requirement already satisfied: imageio>=2.4.1 in /usr/local/lib/python3.9/dist-packages (from scikit-image->basicsr==1.3.5) (2.19.3)\n",
      "Requirement already satisfied: tifffile>=2019.7.26 in /usr/local/lib/python3.9/dist-packages (from scikit-image->basicsr==1.3.5) (2022.5.4)\n",
      "Requirement already satisfied: packaging>=20.0 in /usr/local/lib/python3.9/dist-packages (from scikit-image->basicsr==1.3.5) (21.3)\n",
      "Requirement already satisfied: PyWavelets>=1.1.1 in /usr/local/lib/python3.9/dist-packages (from scikit-image->basicsr==1.3.5) (1.3.0)\n",
      "Requirement already satisfied: grpcio>=1.24.3 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->basicsr==1.3.5) (1.47.0)\n",
      "Requirement already satisfied: google-auth<3,>=1.6.3 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->basicsr==1.3.5) (2.9.0)\n",
      "Requirement already satisfied: setuptools>=41.0.0 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->basicsr==1.3.5) (63.1.0)\n",
      "Requirement already satisfied: wheel>=0.26 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->basicsr==1.3.5) (0.35.1)\n",
      "Requirement already satisfied: protobuf<3.20,>=3.9.2 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->basicsr==1.3.5) (3.19.4)\n",
      "Requirement already satisfied: markdown>=2.6.8 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->basicsr==1.3.5) (3.3.7)\n",
      "Requirement already satisfied: werkzeug>=1.0.1 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->basicsr==1.3.5) (2.2.2)\n",
      "Requirement already satisfied: tensorboard-data-server<0.7.0,>=0.6.0 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->basicsr==1.3.5) (0.6.1)\n",
      "Requirement already satisfied: absl-py>=0.4 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->basicsr==1.3.5) (1.1.0)\n",
      "Requirement already satisfied: tensorboard-plugin-wit>=1.6.0 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->basicsr==1.3.5) (1.8.1)\n",
      "Requirement already satisfied: google-auth-oauthlib<0.5,>=0.4.1 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->basicsr==1.3.5) (0.4.6)\n",
      "Requirement already satisfied: cachetools<6.0,>=2.0.0 in /usr/local/lib/python3.9/dist-packages (from google-auth<3,>=1.6.3->tb-nightly->basicsr==1.3.5) (5.2.0)\n",
      "Requirement already satisfied: pyasn1-modules>=0.2.1 in /usr/local/lib/python3.9/dist-packages (from google-auth<3,>=1.6.3->tb-nightly->basicsr==1.3.5) (0.2.8)\n",
      "Requirement already satisfied: rsa<5,>=3.1.4 in /usr/lib/python3/dist-packages (from google-auth<3,>=1.6.3->tb-nightly->basicsr==1.3.5) (4.0)\n",
      "Requirement already satisfied: six>=1.9.0 in /usr/lib/python3/dist-packages (from google-auth<3,>=1.6.3->tb-nightly->basicsr==1.3.5) (1.14.0)\n",
      "Requirement already satisfied: requests-oauthlib>=0.7.0 in /usr/local/lib/python3.9/dist-packages (from google-auth-oauthlib<0.5,>=0.4.1->tb-nightly->basicsr==1.3.5) (1.3.1)\n",
      "Requirement already satisfied: importlib-metadata>=4.4 in /usr/local/lib/python3.9/dist-packages (from markdown>=2.6.8->tb-nightly->basicsr==1.3.5) (4.12.0)\n",
      "Requirement already satisfied: pyparsing!=3.0.5,>=2.0.2 in /usr/local/lib/python3.9/dist-packages (from packaging>=20.0->scikit-image->basicsr==1.3.5) (3.0.9)\n",
      "Requirement already satisfied: MarkupSafe>=2.1.1 in /usr/local/lib/python3.9/dist-packages (from werkzeug>=1.0.1->tb-nightly->basicsr==1.3.5) (2.1.1)\n",
      "Requirement already satisfied: zipp>=0.5 in /usr/local/lib/python3.9/dist-packages (from importlib-metadata>=4.4->markdown>=2.6.8->tb-nightly->basicsr==1.3.5) (3.8.1)\n",
      "Requirement already satisfied: pyasn1<0.5.0,>=0.4.6 in /usr/local/lib/python3.9/dist-packages (from pyasn1-modules>=0.2.1->google-auth<3,>=1.6.3->tb-nightly->basicsr==1.3.5) (0.4.8)\n",
      "Requirement already satisfied: oauthlib>=3.0.0 in /usr/local/lib/python3.9/dist-packages (from requests-oauthlib>=0.7.0->google-auth-oauthlib<0.5,>=0.4.1->tb-nightly->basicsr==1.3.5) (3.2.0)\n",
      "Building wheels for collected packages: basicsr\n",
      "  Building wheel for basicsr (setup.py) ... \u001b[?25ldone\n",
      "\u001b[?25h  Created wheel for basicsr: filename=basicsr-1.3.5-py3-none-any.whl size=194466 sha256=fd67f04f263a087397e3b3056ed43b59a0ed05c17eebe0f58e6a305edfc9dafb\n",
      "  Stored in directory: /root/.cache/pip/wheels/0c/be/6e/251c52c71371db49e996dedb90875da8e828d65bbd6f7b4e91\n",
      "Successfully built basicsr\n",
      "Installing collected packages: basicsr\n",
      "  Attempting uninstall: basicsr\n",
      "    Found existing installation: basicsr 1.3.4.0\n",
      "    Uninstalling basicsr-1.3.4.0:\n",
      "      Successfully uninstalled basicsr-1.3.4.0\n",
      "Successfully installed basicsr-1.3.5\n",
      "\u001b[33mWARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager. It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv\u001b[0m\u001b[33m\n",
      "\u001b[0mRequirement already satisfied: facexlib in /usr/local/lib/python3.9/dist-packages (0.2.4)\n",
      "Requirement already satisfied: scipy in /usr/local/lib/python3.9/dist-packages (from facexlib) (1.8.1)\n",
      "Requirement already satisfied: torch in /usr/local/lib/python3.9/dist-packages (from facexlib) (1.12.0+cu116)\n",
      "Requirement already satisfied: Pillow in /usr/local/lib/python3.9/dist-packages (from facexlib) (9.2.0)\n",
      "Requirement already satisfied: filterpy in /usr/local/lib/python3.9/dist-packages (from facexlib) (1.4.5)\n",
      "Requirement already satisfied: numpy in /usr/local/lib/python3.9/dist-packages (from facexlib) (1.20.3)\n",
      "Requirement already satisfied: numba in /usr/local/lib/python3.9/dist-packages (from facexlib) (0.56.0)\n",
      "Requirement already satisfied: tqdm in /usr/local/lib/python3.9/dist-packages (from facexlib) (4.64.0)\n",
      "Requirement already satisfied: torchvision in /usr/local/lib/python3.9/dist-packages (from facexlib) (0.13.0+cu116)\n",
      "Requirement already satisfied: opencv-python in /usr/local/lib/python3.9/dist-packages (from facexlib) (4.5.5.64)\n",
      "Requirement already satisfied: matplotlib in /usr/local/lib/python3.9/dist-packages (from filterpy->facexlib) (3.5.2)\n",
      "Requirement already satisfied: setuptools in /usr/local/lib/python3.9/dist-packages (from numba->facexlib) (63.1.0)\n",
      "Requirement already satisfied: llvmlite<0.40,>=0.39.0dev0 in /usr/local/lib/python3.9/dist-packages (from numba->facexlib) (0.39.0)\n",
      "Requirement already satisfied: typing-extensions in /usr/local/lib/python3.9/dist-packages (from torch->facexlib) (4.3.0)\n",
      "Requirement already satisfied: requests in /usr/local/lib/python3.9/dist-packages (from torchvision->facexlib) (2.28.1)\n",
      "Requirement already satisfied: cycler>=0.10 in /usr/local/lib/python3.9/dist-packages (from matplotlib->filterpy->facexlib) (0.11.0)\n",
      "Requirement already satisfied: kiwisolver>=1.0.1 in /usr/local/lib/python3.9/dist-packages (from matplotlib->filterpy->facexlib) (1.4.3)\n",
      "Requirement already satisfied: packaging>=20.0 in /usr/local/lib/python3.9/dist-packages (from matplotlib->filterpy->facexlib) (21.3)\n",
      "Requirement already satisfied: python-dateutil>=2.7 in /usr/local/lib/python3.9/dist-packages (from matplotlib->filterpy->facexlib) (2.8.2)\n",
      "Requirement already satisfied: fonttools>=4.22.0 in /usr/local/lib/python3.9/dist-packages (from matplotlib->filterpy->facexlib) (4.34.4)\n",
      "Requirement already satisfied: pyparsing>=2.2.1 in /usr/local/lib/python3.9/dist-packages (from matplotlib->filterpy->facexlib) (3.0.9)\n",
      "Requirement already satisfied: charset-normalizer<3,>=2 in /usr/local/lib/python3.9/dist-packages (from requests->torchvision->facexlib) (2.1.0)\n",
      "Requirement already satisfied: certifi>=2017.4.17 in /usr/lib/python3/dist-packages (from requests->torchvision->facexlib) (2019.11.28)\n",
      "Requirement already satisfied: urllib3<1.27,>=1.21.1 in /usr/local/lib/python3.9/dist-packages (from requests->torchvision->facexlib) (1.26.10)\n",
      "Requirement already satisfied: idna<4,>=2.5 in /usr/lib/python3/dist-packages (from requests->torchvision->facexlib) (2.8)\n",
      "Requirement already satisfied: six>=1.5 in /usr/lib/python3/dist-packages (from python-dateutil>=2.7->matplotlib->filterpy->facexlib) (1.14.0)\n",
      "\u001b[33mWARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager. It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv\u001b[0m\u001b[33m\n",
      "\u001b[0mRequirement already satisfied: realesrgan==0.2.5 in /usr/local/lib/python3.9/dist-packages (0.2.5.0)\n",
      "Requirement already satisfied: torchvision in /usr/local/lib/python3.9/dist-packages (from realesrgan==0.2.5) (0.13.0+cu116)\n",
      "Requirement already satisfied: numpy in /usr/local/lib/python3.9/dist-packages (from realesrgan==0.2.5) (1.20.3)\n",
      "Requirement already satisfied: facexlib>=0.2.0.3 in /usr/local/lib/python3.9/dist-packages (from realesrgan==0.2.5) (0.2.4)\n",
      "Requirement already satisfied: torch>=1.7 in /usr/local/lib/python3.9/dist-packages (from realesrgan==0.2.5) (1.12.0+cu116)\n",
      "Requirement already satisfied: tqdm in /usr/local/lib/python3.9/dist-packages (from realesrgan==0.2.5) (4.64.0)\n",
      "Requirement already satisfied: basicsr>=1.3.3.11 in /usr/local/lib/python3.9/dist-packages (from realesrgan==0.2.5) (1.3.5)\n",
      "Requirement already satisfied: opencv-python in /usr/local/lib/python3.9/dist-packages (from realesrgan==0.2.5) (4.5.5.64)\n",
      "Requirement already satisfied: Pillow in /usr/local/lib/python3.9/dist-packages (from realesrgan==0.2.5) (9.2.0)\n",
      "Requirement already satisfied: gfpgan>=0.2.1 in /usr/local/lib/python3.9/dist-packages (from realesrgan==0.2.5) (1.3.4)\n",
      "Requirement already satisfied: yapf in /usr/local/lib/python3.9/dist-packages (from basicsr>=1.3.3.11->realesrgan==0.2.5) (0.32.0)\n",
      "Requirement already satisfied: future in /usr/lib/python3/dist-packages (from basicsr>=1.3.3.11->realesrgan==0.2.5) (0.18.2)\n",
      "Requirement already satisfied: requests in /usr/local/lib/python3.9/dist-packages (from basicsr>=1.3.3.11->realesrgan==0.2.5) (2.28.1)\n",
      "Requirement already satisfied: tb-nightly in /usr/local/lib/python3.9/dist-packages (from basicsr>=1.3.3.11->realesrgan==0.2.5) (2.11.0a20220824)\n",
      "Requirement already satisfied: scikit-image in /usr/local/lib/python3.9/dist-packages (from basicsr>=1.3.3.11->realesrgan==0.2.5) (0.19.3)\n",
      "Requirement already satisfied: pyyaml in /usr/local/lib/python3.9/dist-packages (from basicsr>=1.3.3.11->realesrgan==0.2.5) (5.4.1)\n",
      "Requirement already satisfied: scipy in /usr/local/lib/python3.9/dist-packages (from basicsr>=1.3.3.11->realesrgan==0.2.5) (1.8.1)\n",
      "Requirement already satisfied: lmdb in /usr/local/lib/python3.9/dist-packages (from basicsr>=1.3.3.11->realesrgan==0.2.5) (1.3.0)\n",
      "Requirement already satisfied: addict in /usr/local/lib/python3.9/dist-packages (from basicsr>=1.3.3.11->realesrgan==0.2.5) (2.4.0)\n",
      "Requirement already satisfied: numba in /usr/local/lib/python3.9/dist-packages (from facexlib>=0.2.0.3->realesrgan==0.2.5) (0.56.0)\n",
      "Requirement already satisfied: filterpy in /usr/local/lib/python3.9/dist-packages (from facexlib>=0.2.0.3->realesrgan==0.2.5) (1.4.5)\n",
      "Requirement already satisfied: typing-extensions in /usr/local/lib/python3.9/dist-packages (from torch>=1.7->realesrgan==0.2.5) (4.3.0)\n",
      "Requirement already satisfied: matplotlib in /usr/local/lib/python3.9/dist-packages (from filterpy->facexlib>=0.2.0.3->realesrgan==0.2.5) (3.5.2)\n",
      "Requirement already satisfied: llvmlite<0.40,>=0.39.0dev0 in /usr/local/lib/python3.9/dist-packages (from numba->facexlib>=0.2.0.3->realesrgan==0.2.5) (0.39.0)\n",
      "Requirement already satisfied: setuptools in /usr/local/lib/python3.9/dist-packages (from numba->facexlib>=0.2.0.3->realesrgan==0.2.5) (63.1.0)\n",
      "Requirement already satisfied: urllib3<1.27,>=1.21.1 in /usr/local/lib/python3.9/dist-packages (from requests->basicsr>=1.3.3.11->realesrgan==0.2.5) (1.26.10)\n",
      "Requirement already satisfied: charset-normalizer<3,>=2 in /usr/local/lib/python3.9/dist-packages (from requests->basicsr>=1.3.3.11->realesrgan==0.2.5) (2.1.0)\n",
      "Requirement already satisfied: certifi>=2017.4.17 in /usr/lib/python3/dist-packages (from requests->basicsr>=1.3.3.11->realesrgan==0.2.5) (2019.11.28)\n",
      "Requirement already satisfied: idna<4,>=2.5 in /usr/lib/python3/dist-packages (from requests->basicsr>=1.3.3.11->realesrgan==0.2.5) (2.8)\n",
      "Requirement already satisfied: packaging>=20.0 in /usr/local/lib/python3.9/dist-packages (from scikit-image->basicsr>=1.3.3.11->realesrgan==0.2.5) (21.3)\n",
      "Requirement already satisfied: imageio>=2.4.1 in /usr/local/lib/python3.9/dist-packages (from scikit-image->basicsr>=1.3.3.11->realesrgan==0.2.5) (2.19.3)\n",
      "Requirement already satisfied: tifffile>=2019.7.26 in /usr/local/lib/python3.9/dist-packages (from scikit-image->basicsr>=1.3.3.11->realesrgan==0.2.5) (2022.5.4)\n",
      "Requirement already satisfied: PyWavelets>=1.1.1 in /usr/local/lib/python3.9/dist-packages (from scikit-image->basicsr>=1.3.3.11->realesrgan==0.2.5) (1.3.0)\n",
      "Requirement already satisfied: networkx>=2.2 in /usr/local/lib/python3.9/dist-packages (from scikit-image->basicsr>=1.3.3.11->realesrgan==0.2.5) (2.8.4)\n",
      "Requirement already satisfied: absl-py>=0.4 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->basicsr>=1.3.3.11->realesrgan==0.2.5) (1.1.0)\n",
      "Requirement already satisfied: wheel>=0.26 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->basicsr>=1.3.3.11->realesrgan==0.2.5) (0.35.1)\n",
      "Requirement already satisfied: grpcio>=1.24.3 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->basicsr>=1.3.3.11->realesrgan==0.2.5) (1.47.0)\n",
      "Requirement already satisfied: protobuf<3.20,>=3.9.2 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->basicsr>=1.3.3.11->realesrgan==0.2.5) (3.19.4)\n",
      "Requirement already satisfied: google-auth-oauthlib<0.5,>=0.4.1 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->basicsr>=1.3.3.11->realesrgan==0.2.5) (0.4.6)\n",
      "Requirement already satisfied: google-auth<3,>=1.6.3 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->basicsr>=1.3.3.11->realesrgan==0.2.5) (2.9.0)\n",
      "Requirement already satisfied: werkzeug>=1.0.1 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->basicsr>=1.3.3.11->realesrgan==0.2.5) (2.2.2)\n",
      "Requirement already satisfied: tensorboard-plugin-wit>=1.6.0 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->basicsr>=1.3.3.11->realesrgan==0.2.5) (1.8.1)\n",
      "Requirement already satisfied: markdown>=2.6.8 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->basicsr>=1.3.3.11->realesrgan==0.2.5) (3.3.7)\n",
      "Requirement already satisfied: tensorboard-data-server<0.7.0,>=0.6.0 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->basicsr>=1.3.3.11->realesrgan==0.2.5) (0.6.1)\n",
      "Requirement already satisfied: rsa<5,>=3.1.4 in /usr/lib/python3/dist-packages (from google-auth<3,>=1.6.3->tb-nightly->basicsr>=1.3.3.11->realesrgan==0.2.5) (4.0)\n",
      "Requirement already satisfied: cachetools<6.0,>=2.0.0 in /usr/local/lib/python3.9/dist-packages (from google-auth<3,>=1.6.3->tb-nightly->basicsr>=1.3.3.11->realesrgan==0.2.5) (5.2.0)\n",
      "Requirement already satisfied: six>=1.9.0 in /usr/lib/python3/dist-packages (from google-auth<3,>=1.6.3->tb-nightly->basicsr>=1.3.3.11->realesrgan==0.2.5) (1.14.0)\n",
      "Requirement already satisfied: pyasn1-modules>=0.2.1 in /usr/local/lib/python3.9/dist-packages (from google-auth<3,>=1.6.3->tb-nightly->basicsr>=1.3.3.11->realesrgan==0.2.5) (0.2.8)\n",
      "Requirement already satisfied: requests-oauthlib>=0.7.0 in /usr/local/lib/python3.9/dist-packages (from google-auth-oauthlib<0.5,>=0.4.1->tb-nightly->basicsr>=1.3.3.11->realesrgan==0.2.5) (1.3.1)\n",
      "Requirement already satisfied: importlib-metadata>=4.4 in /usr/local/lib/python3.9/dist-packages (from markdown>=2.6.8->tb-nightly->basicsr>=1.3.3.11->realesrgan==0.2.5) (4.12.0)\n",
      "Requirement already satisfied: pyparsing!=3.0.5,>=2.0.2 in /usr/local/lib/python3.9/dist-packages (from packaging>=20.0->scikit-image->basicsr>=1.3.3.11->realesrgan==0.2.5) (3.0.9)\n",
      "Requirement already satisfied: MarkupSafe>=2.1.1 in /usr/local/lib/python3.9/dist-packages (from werkzeug>=1.0.1->tb-nightly->basicsr>=1.3.3.11->realesrgan==0.2.5) (2.1.1)\n",
      "Requirement already satisfied: cycler>=0.10 in /usr/local/lib/python3.9/dist-packages (from matplotlib->filterpy->facexlib>=0.2.0.3->realesrgan==0.2.5) (0.11.0)\n",
      "Requirement already satisfied: fonttools>=4.22.0 in /usr/local/lib/python3.9/dist-packages (from matplotlib->filterpy->facexlib>=0.2.0.3->realesrgan==0.2.5) (4.34.4)\n",
      "Requirement already satisfied: kiwisolver>=1.0.1 in /usr/local/lib/python3.9/dist-packages (from matplotlib->filterpy->facexlib>=0.2.0.3->realesrgan==0.2.5) (1.4.3)\n",
      "Requirement already satisfied: python-dateutil>=2.7 in /usr/local/lib/python3.9/dist-packages (from matplotlib->filterpy->facexlib>=0.2.0.3->realesrgan==0.2.5) (2.8.2)\n",
      "Requirement already satisfied: zipp>=0.5 in /usr/local/lib/python3.9/dist-packages (from importlib-metadata>=4.4->markdown>=2.6.8->tb-nightly->basicsr>=1.3.3.11->realesrgan==0.2.5) (3.8.1)\n",
      "Requirement already satisfied: pyasn1<0.5.0,>=0.4.6 in /usr/local/lib/python3.9/dist-packages (from pyasn1-modules>=0.2.1->google-auth<3,>=1.6.3->tb-nightly->basicsr>=1.3.3.11->realesrgan==0.2.5) (0.4.8)\n",
      "Requirement already satisfied: oauthlib>=3.0.0 in /usr/local/lib/python3.9/dist-packages (from requests-oauthlib>=0.7.0->google-auth-oauthlib<0.5,>=0.4.1->tb-nightly->basicsr>=1.3.3.11->realesrgan==0.2.5) (3.2.0)\n",
      "\u001b[33mWARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager. It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv\u001b[0m\u001b[33m\n",
      "\u001b[0m"
     ]
    }
   ],
   "metadata": {
    "collapsed": false,
    "execution": {
     "iopub.execute_input": "2022-08-24T14:11:07.656684Z",
     "iopub.status.busy": "2022-08-24T14:11:07.656192Z",
     "iopub.status.idle": "2022-08-24T14:11:20.800663Z",
     "shell.execute_reply": "2022-08-24T14:11:20.799941Z",
     "shell.execute_reply.started": "2022-08-24T14:11:07.656644Z"
    },
    "gradient": {
     "editing": false,
     "execution_count": 6,
     "id": "e75afc1a-5a04-4886-a852-8d4311c5651a",
     "kernelId": "a5a7e424-6201-4e2d-8ab7-cb22b6774ff4",
     "source_hidden": false
    },
    "jupyter": {
     "outputs_hidden": false
    }
   }
  },
  {
   "cell_type": "markdown",
   "source": [
    "There is a requirements.txt file which outlines the remaining required installs. run pip install -r to install these packages\n",
    "\n",
    "cv2 is a required package to run GFPGAN, but Gradient requires the headless version of the package to run properly. pip install opencv-python-headless to get a version that runs with Gradient."
   ],
   "metadata": {
    "gradient": {
     "editing": false,
     "id": "31a600f9-6445-4803-a8d0-bcf605391236",
     "kernelId": "a5a7e424-6201-4e2d-8ab7-cb22b6774ff4"
    }
   }
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "source": [
    "!pip install -r requirements.txt\n",
    "!pip install opencv-python==4.5.5.64\n",
    "\n"
   ],
   "outputs": [
    {
     "output_type": "stream",
     "name": "stdout",
     "text": [
      "Requirement already satisfied: torch>=1.7+cpu in /usr/local/lib/python3.9/dist-packages (from -r requirements.txt (line 1)) (1.12.0+cu116)\n",
      "Requirement already satisfied: numpy<1.21 in /usr/local/lib/python3.9/dist-packages (from -r requirements.txt (line 2)) (1.20.3)\n",
      "Requirement already satisfied: torchvision in /usr/local/lib/python3.9/dist-packages (from -r requirements.txt (line 3)) (0.13.0+cu116)\n",
      "Requirement already satisfied: scipy in /usr/local/lib/python3.9/dist-packages (from -r requirements.txt (line 4)) (1.8.1)\n",
      "Requirement already satisfied: tqdm in /usr/local/lib/python3.9/dist-packages (from -r requirements.txt (line 5)) (4.64.0)\n",
      "Requirement already satisfied: lmdb in /usr/local/lib/python3.9/dist-packages (from -r requirements.txt (line 6)) (1.3.0)\n",
      "Requirement already satisfied: pyyaml in /usr/local/lib/python3.9/dist-packages (from -r requirements.txt (line 7)) (5.4.1)\n",
      "Requirement already satisfied: tb-nightly in /usr/local/lib/python3.9/dist-packages (from -r requirements.txt (line 8)) (2.11.0a20220824)\n",
      "Requirement already satisfied: yapf in /usr/local/lib/python3.9/dist-packages (from -r requirements.txt (line 9)) (0.32.0)\n",
      "Collecting Flask\n",
      "  Downloading Flask-2.2.2-py3-none-any.whl (101 kB)\n",
      "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m101.5/101.5 kB\u001b[0m \u001b[31m10.6 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hRequirement already satisfied: typing-extensions in /usr/local/lib/python3.9/dist-packages (from torch>=1.7+cpu->-r requirements.txt (line 1)) (4.3.0)\n",
      "Requirement already satisfied: pillow!=8.3.*,>=5.3.0 in /usr/local/lib/python3.9/dist-packages (from torchvision->-r requirements.txt (line 3)) (9.2.0)\n",
      "Requirement already satisfied: requests in /usr/local/lib/python3.9/dist-packages (from torchvision->-r requirements.txt (line 3)) (2.28.1)\n",
      "Requirement already satisfied: google-auth<3,>=1.6.3 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->-r requirements.txt (line 8)) (2.9.0)\n",
      "Requirement already satisfied: werkzeug>=1.0.1 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->-r requirements.txt (line 8)) (2.1.2)\n",
      "Requirement already satisfied: wheel>=0.26 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->-r requirements.txt (line 8)) (0.35.1)\n",
      "Requirement already satisfied: setuptools>=41.0.0 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->-r requirements.txt (line 8)) (63.1.0)\n",
      "Requirement already satisfied: absl-py>=0.4 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->-r requirements.txt (line 8)) (1.1.0)\n",
      "Requirement already satisfied: grpcio>=1.24.3 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->-r requirements.txt (line 8)) (1.47.0)\n",
      "Requirement already satisfied: markdown>=2.6.8 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->-r requirements.txt (line 8)) (3.3.7)\n",
      "Requirement already satisfied: tensorboard-plugin-wit>=1.6.0 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->-r requirements.txt (line 8)) (1.8.1)\n",
      "Requirement already satisfied: protobuf<3.20,>=3.9.2 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->-r requirements.txt (line 8)) (3.19.4)\n",
      "Requirement already satisfied: tensorboard-data-server<0.7.0,>=0.6.0 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->-r requirements.txt (line 8)) (0.6.1)\n",
      "Requirement already satisfied: google-auth-oauthlib<0.5,>=0.4.1 in /usr/local/lib/python3.9/dist-packages (from tb-nightly->-r requirements.txt (line 8)) (0.4.6)\n",
      "Requirement already satisfied: importlib-metadata>=3.6.0 in /usr/local/lib/python3.9/dist-packages (from Flask->-r requirements.txt (line 10)) (4.12.0)\n",
      "Collecting werkzeug>=1.0.1\n",
      "  Downloading Werkzeug-2.2.2-py3-none-any.whl (232 kB)\n",
      "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m232.7/232.7 kB\u001b[0m \u001b[31m38.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hRequirement already satisfied: Jinja2>=3.0 in /usr/local/lib/python3.9/dist-packages (from Flask->-r requirements.txt (line 10)) (3.1.2)\n",
      "Collecting itsdangerous>=2.0\n",
      "  Downloading itsdangerous-2.1.2-py3-none-any.whl (15 kB)\n",
      "Requirement already satisfied: click>=8.0 in /usr/local/lib/python3.9/dist-packages (from Flask->-r requirements.txt (line 10)) (8.1.3)\n",
      "Requirement already satisfied: rsa<5,>=3.1.4 in /usr/lib/python3/dist-packages (from google-auth<3,>=1.6.3->tb-nightly->-r requirements.txt (line 8)) (4.0)\n",
      "Requirement already satisfied: six>=1.9.0 in /usr/lib/python3/dist-packages (from google-auth<3,>=1.6.3->tb-nightly->-r requirements.txt (line 8)) (1.14.0)\n",
      "Requirement already satisfied: cachetools<6.0,>=2.0.0 in /usr/local/lib/python3.9/dist-packages (from google-auth<3,>=1.6.3->tb-nightly->-r requirements.txt (line 8)) (5.2.0)\n",
      "Requirement already satisfied: pyasn1-modules>=0.2.1 in /usr/local/lib/python3.9/dist-packages (from google-auth<3,>=1.6.3->tb-nightly->-r requirements.txt (line 8)) (0.2.8)\n",
      "Requirement already satisfied: requests-oauthlib>=0.7.0 in /usr/local/lib/python3.9/dist-packages (from google-auth-oauthlib<0.5,>=0.4.1->tb-nightly->-r requirements.txt (line 8)) (1.3.1)\n",
      "Requirement already satisfied: zipp>=0.5 in /usr/local/lib/python3.9/dist-packages (from importlib-metadata>=3.6.0->Flask->-r requirements.txt (line 10)) (3.8.1)\n",
      "Requirement already satisfied: MarkupSafe>=2.0 in /usr/local/lib/python3.9/dist-packages (from Jinja2>=3.0->Flask->-r requirements.txt (line 10)) (2.1.1)\n",
      "Requirement already satisfied: idna<4,>=2.5 in /usr/lib/python3/dist-packages (from requests->torchvision->-r requirements.txt (line 3)) (2.8)\n",
      "Requirement already satisfied: certifi>=2017.4.17 in /usr/lib/python3/dist-packages (from requests->torchvision->-r requirements.txt (line 3)) (2019.11.28)\n",
      "Requirement already satisfied: charset-normalizer<3,>=2 in /usr/local/lib/python3.9/dist-packages (from requests->torchvision->-r requirements.txt (line 3)) (2.1.0)\n",
      "Requirement already satisfied: urllib3<1.27,>=1.21.1 in /usr/local/lib/python3.9/dist-packages (from requests->torchvision->-r requirements.txt (line 3)) (1.26.10)\n",
      "Requirement already satisfied: pyasn1<0.5.0,>=0.4.6 in /usr/local/lib/python3.9/dist-packages (from pyasn1-modules>=0.2.1->google-auth<3,>=1.6.3->tb-nightly->-r requirements.txt (line 8)) (0.4.8)\n",
      "Requirement already satisfied: oauthlib>=3.0.0 in /usr/local/lib/python3.9/dist-packages (from requests-oauthlib>=0.7.0->google-auth-oauthlib<0.5,>=0.4.1->tb-nightly->-r requirements.txt (line 8)) (3.2.0)\n",
      "Installing collected packages: werkzeug, itsdangerous, Flask\n",
      "  Attempting uninstall: werkzeug\n",
      "    Found existing installation: Werkzeug 2.1.2\n",
      "    Uninstalling Werkzeug-2.1.2:\n",
      "      Successfully uninstalled Werkzeug-2.1.2\n",
      "Successfully installed Flask-2.2.2 itsdangerous-2.1.2 werkzeug-2.2.2\n",
      "\u001b[33mWARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager. It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv\u001b[0m\u001b[33m\n",
      "\u001b[0mCollecting opencv-python==4.5.5.64\n",
      "  Downloading opencv_python-4.5.5.64-cp36-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (60.5 MB)\n",
      "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m60.5/60.5 MB\u001b[0m \u001b[31m24.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m00:01\u001b[0m00:01\u001b[0m\n",
      "\u001b[?25hRequirement already satisfied: numpy>=1.17.3 in /usr/local/lib/python3.9/dist-packages (from opencv-python==4.5.5.64) (1.20.3)\n",
      "Installing collected packages: opencv-python\n",
      "  Attempting uninstall: opencv-python\n",
      "    Found existing installation: opencv-python 4.6.0.66\n",
      "    Uninstalling opencv-python-4.6.0.66:\n",
      "      Successfully uninstalled opencv-python-4.6.0.66\n",
      "Successfully installed opencv-python-4.5.5.64\n",
      "\u001b[33mWARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager. It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv\u001b[0m\u001b[33m\n",
      "\u001b[0m"
     ]
    }
   ],
   "metadata": {
    "collapsed": false,
    "execution": {
     "iopub.execute_input": "2022-08-24T14:06:48.233152Z",
     "iopub.status.busy": "2022-08-24T14:06:48.232832Z",
     "iopub.status.idle": "2022-08-24T14:06:57.642818Z",
     "shell.execute_reply": "2022-08-24T14:06:57.642096Z",
     "shell.execute_reply.started": "2022-08-24T14:06:48.233124Z"
    },
    "gradient": {
     "editing": false,
     "execution_count": 3,
     "id": "dd798b3f-7b8a-453b-9227-b4d26b5ce8cd",
     "kernelId": "a5a7e424-6201-4e2d-8ab7-cb22b6774ff4",
     "source_hidden": false
    },
    "jupyter": {
     "outputs_hidden": false
    }
   }
  },
  {
   "cell_type": "markdown",
   "source": [
    "## Don't skip this!\n",
    "Run the following in the terminal\n",
    "`apt-get update && apt-get install libgl1`\n"
   ],
   "metadata": {}
  },
  {
   "cell_type": "markdown",
   "source": [
    "Run the setup.py develop function to complete the environment setup, and then download the pretrained model for the demo using wget"
   ],
   "metadata": {
    "gradient": {
     "editing": false,
     "id": "062c79cc-2dde-492e-80fb-8e9b4c1b2afb",
     "kernelId": "a5a7e424-6201-4e2d-8ab7-cb22b6774ff4"
    }
   }
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "source": [
    "!apt-get update -y\n",
    "!apt-get install libgl1 -y"
   ],
   "outputs": [
    {
     "output_type": "stream",
     "name": "stdout",
     "text": [
      "Get:1 http://security.ubuntu.com/ubuntu focal-security InRelease [114 kB]\n",
      "Get:2 https://deb.nodesource.com/node_16.x focal InRelease [4583 B]            \n",
      "Get:3 http://ppa.launchpad.net/deadsnakes/ppa/ubuntu focal InRelease [18.1 kB] \n",
      "Get:4 http://security.ubuntu.com/ubuntu focal-security/multiverse amd64 Packages [27.5 kB]\n",
      "Get:5 http://security.ubuntu.com/ubuntu focal-security/restricted amd64 Packages [1462 kB]\n",
      "Get:6 http://security.ubuntu.com/ubuntu focal-security/main amd64 Packages [2092 kB]\n",
      "Get:7 http://archive.ubuntu.com/ubuntu focal InRelease [265 kB]                \n",
      "Get:8 https://deb.nodesource.com/node_16.x focal/main amd64 Packages [775 B]   \n",
      "Get:9 http://security.ubuntu.com/ubuntu focal-security/universe amd64 Packages [889 kB]\n",
      "Get:10 http://ppa.launchpad.net/deadsnakes/ppa/ubuntu focal/main amd64 Packages [26.6 kB]\n",
      "Get:11 http://archive.ubuntu.com/ubuntu focal-updates InRelease [114 kB]\n",
      "Get:12 http://archive.ubuntu.com/ubuntu focal-backports InRelease [108 kB]\n",
      "Get:13 http://archive.ubuntu.com/ubuntu focal/universe amd64 Packages [11.3 MB]\n",
      "Get:14 http://archive.ubuntu.com/ubuntu focal/restricted amd64 Packages [33.4 kB]\n",
      "Get:15 http://archive.ubuntu.com/ubuntu focal/multiverse amd64 Packages [177 kB]\n",
      "Get:16 http://archive.ubuntu.com/ubuntu focal/main amd64 Packages [1275 kB]\n",
      "Get:17 http://archive.ubuntu.com/ubuntu focal-updates/main amd64 Packages [2547 kB]\n",
      "Get:18 http://archive.ubuntu.com/ubuntu focal-updates/multiverse amd64 Packages [30.2 kB]\n",
      "Get:19 http://archive.ubuntu.com/ubuntu focal-updates/restricted amd64 Packages [1572 kB]\n",
      "Get:20 http://archive.ubuntu.com/ubuntu focal-updates/universe amd64 Packages [1183 kB]\n",
      "Get:21 http://archive.ubuntu.com/ubuntu focal-backports/main amd64 Packages [55.1 kB]\n",
      "Get:22 http://archive.ubuntu.com/ubuntu focal-backports/universe amd64 Packages [27.4 kB]\n",
      "Fetched 23.4 MB in 2s (10.4 MB/s)                             \n",
      "Reading package lists... Done\n",
      "Reading package lists... Done\n",
      "Building dependency tree       \n",
      "Reading state information... Done\n",
      "libgl1 is already the newest version (1.3.2-1~ubuntu0.20.04.2).\n",
      "libgl1 set to manually installed.\n",
      "0 upgraded, 0 newly installed, 0 to remove and 47 not upgraded.\n"
     ]
    }
   ],
   "metadata": {
    "execution": {
     "iopub.execute_input": "2022-08-24T14:07:13.946738Z",
     "iopub.status.busy": "2022-08-24T14:07:13.946018Z",
     "iopub.status.idle": "2022-08-24T14:07:19.798611Z",
     "shell.execute_reply": "2022-08-24T14:07:19.797584Z",
     "shell.execute_reply.started": "2022-08-24T14:07:13.946703Z"
    }
   }
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "source": [
    "!python setup.py develop"
   ],
   "outputs": [
    {
     "output_type": "stream",
     "name": "stdout",
     "text": [
      "running develop\n",
      "/usr/local/lib/python3.9/dist-packages/setuptools/command/easy_install.py:144: EasyInstallDeprecationWarning: easy_install command is deprecated. Use build and pip and other standards-based tools.\n",
      "  warnings.warn(\n",
      "/usr/local/lib/python3.9/dist-packages/setuptools/command/install.py:34: SetuptoolsDeprecationWarning: setup.py install is deprecated. Use build and pip and other standards-based tools.\n",
      "  warnings.warn(\n",
      "running egg_info\n",
      "creating gfpgan.egg-info\n",
      "writing gfpgan.egg-info/PKG-INFO\n",
      "writing dependency_links to gfpgan.egg-info/dependency_links.txt\n",
      "writing requirements to gfpgan.egg-info/requires.txt\n",
      "writing top-level names to gfpgan.egg-info/top_level.txt\n",
      "writing manifest file 'gfpgan.egg-info/SOURCES.txt'\n",
      "reading manifest file 'gfpgan.egg-info/SOURCES.txt'\n",
      "reading manifest template 'MANIFEST.in'\n",
      "warning: no files found matching 'assets/*'\n",
      "warning: no files found matching 'inputs/*'\n",
      "adding license file 'LICENSE'\n",
      "writing manifest file 'gfpgan.egg-info/SOURCES.txt'\n",
      "running build_ext\n",
      "Creating /usr/lib/python3.9/site-packages/gfpgan.egg-link (link to .)\n",
      "Adding gfpgan 0.2.3 to easy-install.pth file\n",
      "\n",
      "Installed /notebooks\n",
      "Processing dependencies for gfpgan==0.2.3\n",
      "Searching for Flask==2.2.2\n",
      "Best match: Flask 2.2.2\n",
      "Adding Flask 2.2.2 to easy-install.pth file\n",
      "Installing flask script to /usr/bin\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for yapf==0.32.0\n",
      "Best match: yapf 0.32.0\n",
      "Adding yapf 0.32.0 to easy-install.pth file\n",
      "Installing yapf script to /usr/bin\n",
      "Installing yapf-diff script to /usr/bin\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for tb-nightly==2.11.0a20220824\n",
      "Best match: tb-nightly 2.11.0a20220824\n",
      "Adding tb-nightly 2.11.0a20220824 to easy-install.pth file\n",
      "Installing tensorboard script to /usr/bin\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for PyYAML==5.4.1\n",
      "Best match: PyYAML 5.4.1\n",
      "Adding PyYAML 5.4.1 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for lmdb==1.3.0\n",
      "Best match: lmdb 1.3.0\n",
      "Adding lmdb 1.3.0 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for tqdm==4.64.0\n",
      "Best match: tqdm 4.64.0\n",
      "Adding tqdm 4.64.0 to easy-install.pth file\n",
      "Installing tqdm script to /usr/bin\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for scipy==1.8.1\n",
      "Best match: scipy 1.8.1\n",
      "Adding scipy 1.8.1 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for torchvision==0.13.0+cu116\n",
      "Best match: torchvision 0.13.0+cu116\n",
      "Adding torchvision 0.13.0+cu116 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for numpy==1.20.3\n",
      "Best match: numpy 1.20.3\n",
      "Adding numpy 1.20.3 to easy-install.pth file\n",
      "Installing f2py script to /usr/bin\n",
      "Installing f2py3 script to /usr/bin\n",
      "Installing f2py3.9 script to /usr/bin\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for torch==1.12.0+cu116\n",
      "Best match: torch 1.12.0+cu116\n",
      "Adding torch 1.12.0+cu116 to easy-install.pth file\n",
      "Installing convert-caffe2-to-onnx script to /usr/bin\n",
      "Installing convert-onnx-to-caffe2 script to /usr/bin\n",
      "Installing torchrun script to /usr/bin\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for importlib-metadata==4.12.0\n",
      "Best match: importlib-metadata 4.12.0\n",
      "Adding importlib-metadata 4.12.0 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for click==8.1.3\n",
      "Best match: click 8.1.3\n",
      "Adding click 8.1.3 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for itsdangerous==2.1.2\n",
      "Best match: itsdangerous 2.1.2\n",
      "Adding itsdangerous 2.1.2 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for Jinja2==3.1.2\n",
      "Best match: Jinja2 3.1.2\n",
      "Adding Jinja2 3.1.2 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for Werkzeug==2.2.2\n",
      "Best match: Werkzeug 2.2.2\n",
      "Adding Werkzeug 2.2.2 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for wheel==0.35.1\n",
      "Best match: wheel 0.35.1\n",
      "Adding wheel 0.35.1 to easy-install.pth file\n",
      "Installing wheel script to /usr/bin\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for tensorboard-plugin-wit==1.8.1\n",
      "Best match: tensorboard-plugin-wit 1.8.1\n",
      "Adding tensorboard-plugin-wit 1.8.1 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for tensorboard-data-server==0.6.1\n",
      "Best match: tensorboard-data-server 0.6.1\n",
      "Adding tensorboard-data-server 0.6.1 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for setuptools==63.1.0\n",
      "Best match: setuptools 63.1.0\n",
      "Adding setuptools 63.1.0 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for requests==2.28.1\n",
      "Best match: requests 2.28.1\n",
      "Adding requests 2.28.1 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for protobuf==3.19.4\n",
      "Best match: protobuf 3.19.4\n",
      "Adding protobuf 3.19.4 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for Markdown==3.3.7\n",
      "Best match: Markdown 3.3.7\n",
      "Adding Markdown 3.3.7 to easy-install.pth file\n",
      "Installing markdown_py script to /usr/bin\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for google-auth-oauthlib==0.4.6\n",
      "Best match: google-auth-oauthlib 0.4.6\n",
      "Adding google-auth-oauthlib 0.4.6 to easy-install.pth file\n",
      "Installing google-oauthlib-tool script to /usr/bin\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for google-auth==2.9.0\n",
      "Best match: google-auth 2.9.0\n",
      "Adding google-auth 2.9.0 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for grpcio==1.47.0\n",
      "Best match: grpcio 1.47.0\n",
      "Adding grpcio 1.47.0 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for absl-py==1.1.0\n",
      "Best match: absl-py 1.1.0\n",
      "Adding absl-py 1.1.0 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for Pillow==9.2.0\n",
      "Best match: Pillow 9.2.0\n",
      "Adding Pillow 9.2.0 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for typing-extensions==4.3.0\n",
      "Best match: typing-extensions 4.3.0\n",
      "Adding typing-extensions 4.3.0 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for zipp==3.8.1\n",
      "Best match: zipp 3.8.1\n",
      "Adding zipp 3.8.1 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for MarkupSafe==2.1.1\n",
      "Best match: MarkupSafe 2.1.1\n",
      "Adding MarkupSafe 2.1.1 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for certifi==2019.11.28\n",
      "Best match: certifi 2019.11.28\n",
      "Adding certifi 2019.11.28 to easy-install.pth file\n",
      "\n",
      "Using /usr/lib/python3/dist-packages\n",
      "Searching for urllib3==1.26.10\n",
      "Best match: urllib3 1.26.10\n",
      "Adding urllib3 1.26.10 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for idna==2.8\n",
      "Best match: idna 2.8\n",
      "Adding idna 2.8 to easy-install.pth file\n",
      "\n",
      "Using /usr/lib/python3/dist-packages\n",
      "Searching for charset-normalizer==2.1.0\n",
      "Best match: charset-normalizer 2.1.0\n",
      "Adding charset-normalizer 2.1.0 to easy-install.pth file\n",
      "Installing normalizer script to /usr/bin\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for requests-oauthlib==1.3.1\n",
      "Best match: requests-oauthlib 1.3.1\n",
      "Adding requests-oauthlib 1.3.1 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for rsa==4.0\n",
      "Best match: rsa 4.0\n",
      "Adding rsa 4.0 to easy-install.pth file\n",
      "Installing pyrsa-decrypt script to /usr/bin\n",
      "Installing pyrsa-encrypt script to /usr/bin\n",
      "Installing pyrsa-keygen script to /usr/bin\n",
      "Installing pyrsa-priv2pub script to /usr/bin\n",
      "Installing pyrsa-sign script to /usr/bin\n",
      "Installing pyrsa-verify script to /usr/bin\n",
      "\n",
      "Using /usr/lib/python3/dist-packages\n",
      "Searching for six==1.14.0\n",
      "Best match: six 1.14.0\n",
      "Adding six 1.14.0 to easy-install.pth file\n",
      "\n",
      "Using /usr/lib/python3/dist-packages\n",
      "Searching for pyasn1-modules==0.2.8\n",
      "Best match: pyasn1-modules 0.2.8\n",
      "Adding pyasn1-modules 0.2.8 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for cachetools==5.2.0\n",
      "Best match: cachetools 5.2.0\n",
      "Adding cachetools 5.2.0 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for oauthlib==3.2.0\n",
      "Best match: oauthlib 3.2.0\n",
      "Adding oauthlib 3.2.0 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Searching for pyasn1==0.4.8\n",
      "Best match: pyasn1 0.4.8\n",
      "Adding pyasn1 0.4.8 to easy-install.pth file\n",
      "\n",
      "Using /usr/local/lib/python3.9/dist-packages\n",
      "Finished processing dependencies for gfpgan==0.2.3\n"
     ]
    }
   ],
   "metadata": {
    "collapsed": false,
    "execution": {
     "iopub.execute_input": "2022-08-24T14:07:26.899128Z",
     "iopub.status.busy": "2022-08-24T14:07:26.898795Z",
     "iopub.status.idle": "2022-08-24T14:07:29.073959Z",
     "shell.execute_reply": "2022-08-24T14:07:29.072787Z",
     "shell.execute_reply.started": "2022-08-24T14:07:26.899101Z"
    },
    "gradient": {
     "editing": false,
     "execution_count": 4,
     "id": "71f199f1-71e6-475c-a308-48869e797b9a",
     "kernelId": "a5a7e424-6201-4e2d-8ab7-cb22b6774ff4",
     "source_hidden": false
    },
    "jupyter": {
     "outputs_hidden": false
    }
   }
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "source": [
    "!wget https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth -P experiments/pretrained_models\n"
   ],
   "outputs": [
    {
     "output_type": "stream",
     "name": "stdout",
     "text": [
      "--2022-08-24 14:07:33--  https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth\n",
      "Resolving github.com (github.com)... 140.82.112.4\n",
      "Connecting to github.com (github.com)|140.82.112.4|:443... connected.\n",
      "HTTP request sent, awaiting response... 302 Found\n",
      "Location: https://objects.githubusercontent.com/github-production-release-asset-2e65be/349321229/e9847322-b8b1-4ec2-9620-5146eb8a9e4b?X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Credential=AKIAIWNJYAX4CSVEH53A%2F20220824%2Fus-east-1%2Fs3%2Faws4_request&X-Amz-Date=20220824T140733Z&X-Amz-Expires=300&X-Amz-Signature=c6de676dd7992601049ff735db0da1f95622181ab1180316317f9c1ded4c92ca&X-Amz-SignedHeaders=host&actor_id=0&key_id=0&repo_id=349321229&response-content-disposition=attachment%3B%20filename%3DGFPGANv1.3.pth&response-content-type=application%2Foctet-stream [following]\n",
      "--2022-08-24 14:07:33--  https://objects.githubusercontent.com/github-production-release-asset-2e65be/349321229/e9847322-b8b1-4ec2-9620-5146eb8a9e4b?X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Credential=AKIAIWNJYAX4CSVEH53A%2F20220824%2Fus-east-1%2Fs3%2Faws4_request&X-Amz-Date=20220824T140733Z&X-Amz-Expires=300&X-Amz-Signature=c6de676dd7992601049ff735db0da1f95622181ab1180316317f9c1ded4c92ca&X-Amz-SignedHeaders=host&actor_id=0&key_id=0&repo_id=349321229&response-content-disposition=attachment%3B%20filename%3DGFPGANv1.3.pth&response-content-type=application%2Foctet-stream\n",
      "Resolving objects.githubusercontent.com (objects.githubusercontent.com)... 185.199.110.133, 185.199.109.133, 185.199.111.133, ...\n",
      "Connecting to objects.githubusercontent.com (objects.githubusercontent.com)|185.199.110.133|:443... connected.\n",
      "HTTP request sent, awaiting response... 200 OK\n",
      "Length: 348632874 (332M) [application/octet-stream]\n",
      "Saving to: ‘experiments/pretrained_models/GFPGANv1.3.pth’\n",
      "\n",
      "GFPGANv1.3.pth      100%[===================>] 332.48M  93.3MB/s    in 3.7s    \n",
      "\n",
      "2022-08-24 14:07:37 (90.0 MB/s) - ‘experiments/pretrained_models/GFPGANv1.3.pth’ saved [348632874/348632874]\n",
      "\n"
     ]
    }
   ],
   "metadata": {
    "collapsed": false,
    "execution": {
     "iopub.execute_input": "2022-08-24T14:07:32.887516Z",
     "iopub.status.busy": "2022-08-24T14:07:32.886633Z",
     "iopub.status.idle": "2022-08-24T14:07:37.610235Z",
     "shell.execute_reply": "2022-08-24T14:07:37.609176Z",
     "shell.execute_reply.started": "2022-08-24T14:07:32.887480Z"
    },
    "gradient": {
     "editing": false,
     "execution_count": 5,
     "id": "00e62a06-6504-4afb-a8c9-de44163ec954",
     "kernelId": "a5a7e424-6201-4e2d-8ab7-cb22b6774ff4",
     "source_hidden": false
    },
    "jupyter": {
     "outputs_hidden": false
    }
   }
  },
  {
   "cell_type": "markdown",
   "source": [
    "# Demo\n",
    "\n",
    "For this demonstration, we are going to be using a pretrained version of the model made available by the original authors. \n",
    "\n",
    "I have provided a sample photo in the directory out to be used with this pretrained network. Feel free to upload your own photos to the inputs/whole_imgs directory to try it out!"
   ],
   "metadata": {
    "gradient": {
     "editing": false,
     "id": "1fdbec13-16a3-44de-9373-9a021526a7c5",
     "kernelId": "a5a7e424-6201-4e2d-8ab7-cb22b6774ff4"
    }
   }
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "source": [
    "!python inference_gfpgan.py -i inputs/whole_imgs -o results -v 1.3 -s 2\n"
   ],
   "outputs": [
    {
     "output_type": "stream",
     "name": "stdout",
     "text": [
      "Downloading: \"https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.1/RealESRGAN_x2plus.pth\" to /usr/local/lib/python3.9/dist-packages/realesrgan/weights/RealESRGAN_x2plus.pth\n",
      "\n",
      "100%|███████████████████████████████████████| 64.0M/64.0M [00:00<00:00, 111MB/s]\n",
      "/usr/local/lib/python3.9/dist-packages/torchvision/models/_utils.py:208: UserWarning: The parameter 'pretrained' is deprecated since 0.13 and will be removed in 0.15, please use 'weights' instead.\n",
      "  warnings.warn(\n",
      "/usr/local/lib/python3.9/dist-packages/torchvision/models/_utils.py:223: UserWarning: Arguments other than a weight enum or `None` for 'weights' are deprecated since 0.13 and will be removed in 0.15. The current behavior is equivalent to passing `weights=None`.\n",
      "  warnings.warn(msg)\n",
      "Downloading: \"https://github.com/xinntao/facexlib/releases/download/v0.1.0/detection_Resnet50_Final.pth\" to /usr/local/lib/python3.9/dist-packages/facexlib/weights/detection_Resnet50_Final.pth\n",
      "\n",
      "100%|█████████████████████████████████████████| 104M/104M [00:00<00:00, 126MB/s]\n",
      "Downloading: \"https://github.com/xinntao/facexlib/releases/download/v0.2.2/parsing_parsenet.pth\" to /usr/local/lib/python3.9/dist-packages/facexlib/weights/parsing_parsenet.pth\n",
      "\n",
      "100%|███████████████████████████████████████| 81.4M/81.4M [00:00<00:00, 312MB/s]\n",
      "Processing 00.jpg ...\n",
      "\tTile 1/20\n",
      "\tTile 2/20\n",
      "\tTile 3/20\n",
      "\tTile 4/20\n",
      "\tTile 5/20\n",
      "\tTile 6/20\n",
      "\tTile 7/20\n",
      "\tTile 8/20\n",
      "\tTile 9/20\n",
      "\tTile 10/20\n",
      "\tTile 11/20\n",
      "\tTile 12/20\n",
      "\tTile 13/20\n",
      "\tTile 14/20\n",
      "\tTile 15/20\n",
      "\tTile 16/20\n",
      "\tTile 17/20\n",
      "\tTile 18/20\n",
      "\tTile 19/20\n",
      "\tTile 20/20\n",
      "Processing 10045.png ...\n",
      "\tTile 1/6\n",
      "\tTile 2/6\n",
      "\tTile 3/6\n",
      "\tTile 4/6\n",
      "\tTile 5/6\n",
      "\tTile 6/6\n",
      "Processing Blake_Lively.jpg ...\n",
      "\tTile 1/4\n",
      "\tTile 2/4\n",
      "\tTile 3/4\n",
      "\tTile 4/4\n",
      "Results are in the [results] folder.\n"
     ]
    }
   ],
   "metadata": {
    "collapsed": false,
    "execution": {
     "iopub.execute_input": "2022-08-24T14:11:26.703287Z",
     "iopub.status.busy": "2022-08-24T14:11:26.702358Z",
     "iopub.status.idle": "2022-08-24T14:11:46.885620Z",
     "shell.execute_reply": "2022-08-24T14:11:46.884767Z",
     "shell.execute_reply.started": "2022-08-24T14:11:26.703253Z"
    },
    "gradient": {
     "editing": false,
     "execution_count": 7,
     "id": "fc9c3ff9-466e-4d6a-884f-4368328bfb93",
     "kernelId": "a5a7e424-6201-4e2d-8ab7-cb22b6774ff4",
     "source_hidden": false
    },
    "jupyter": {
     "outputs_hidden": false
    }
   }
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "source": [],
   "outputs": [],
   "metadata": {}
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.9.13"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}