Portrait_Painter_Z_(CPU_or_GPU).ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "name": "Portrait Painter Z (CPU or GPU)",
      "provenance": [],
      "collapsed_sections": [],
      "include_colab_link": true
    },
    "kernelspec": {
      "display_name": "Python 3",
      "name": "python3"
    },
    "accelerator": "GPU"
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/MrPalais/U-2-Net_Z/blob/master/Portrait_Painter_Z_(CPU_or_GPU).ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "6IxcdscIUd2h"
      },
      "source": [
        "# <b><font color=\"black\" size=\"+4\">Sketch portrait for photo/video</font></b>\n",
        "\n",
        "<b><font color=\"black\" size=\"+2\">Основан на:</font></b>\n",
        "\n",
        "**GitHub репозитории**: [U-2-Net](https://github.com/NathanUA/U-2-Net)\n",
        "\n",
        "**Статья**: [U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection](https://arxiv.org/pdf/2005.09007.pdf)\n",
        "\n",
        "**Авторы**: [Xuebin Qin](https://webdocs.cs.ualberta.ca/~xuebin/), [Zichen Zhang](https://webdocs.cs.ualberta.ca/~zichen2/), [Chenyang Huang](https://chenyangh.com/), [Masood Dehghan](https://sites.google.com/view/masooddehghan), [Osmar R. Zaiane](http://webdocs.cs.ualberta.ca/~zaiane/), [Martin Jagersand](https://webdocs.cs.ualberta.ca/~jag/)\n",
        "\n",
        "<b><font color=\"black\" size=\"+2\">Colab собрал:</font></b>\n",
        "\n",
        "GitHub: [@tg-bomze](https://github.com/tg-bomze),\n",
        "Telegram: [@bomze](https://t.me/bomze),\n",
        "Twitter: [@tg_bomze](https://twitter.com/tg_bomze).\n",
        "\n",
        "```\n",
        "Version modified bt MrPalais to use CPU or GPU\n",
        "You can mount your google drive and specify /content/drive/MyDrive/.../myvideo.mp4 as input\n",
        "```\n"
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "x5YWPQE7TBmg",
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "cellView": "form",
        "outputId": "ee8e3515-e298-4be8-bb5e-662486646161"
      },
      "source": [
        "#@title <b><font color=\"red\" size=\"+3\">←</font><font color=\"black\" size=\"+3\">Init files and gdrive</font></b>\n",
        "\n",
        "# if you need your drive\n",
        "from google.colab import drive \n",
        "drive.mount('/content/drive')\n",
        "\n",
        "!nvidia-smi\n",
        "\n",
        "# in case if you need to install the CPU version of torch\n",
        "#!pip install torch==1.9.0+cpu torchvision==0.10.0+cpu torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html\n",
        "!pip install ffmpeg\n",
        "!pip install youtube-dl\n",
        "\n",
        "from IPython.display import HTML, clear_output\n",
        "from IPython import display as ipythondisplay\n",
        "from google.colab.patches import cv2_imshow\n",
        "import moviepy.editor as mpy\n",
        "from moviepy.video.io.ffmpeg_writer import FFMPEG_VideoWriter\n",
        "from google.colab import files\n",
        "import numpy as np\n",
        "import youtube_dl\n",
        "from PIL import Image\n",
        "import matplotlib.pyplot as plt\n",
        "import warnings\n",
        "import ffmpeg\n",
        "import cv2\n",
        "from base64 import b64encode\n",
        "import os\n",
        "import io\n",
        "\n",
        "def display_video(video_path, width=640, clear=True):\n",
        "  vid_data = open(video_path,'rb').read()\n",
        "  vid_url = 'data:video/mp4;base64,' + b64encode(vid_data).decode()\n",
        "  if clear: clear_output()\n",
        "  return HTML(f\"\"\"\n",
        "  <video width={width} controls>\n",
        "    <source src={vid_url} type=\"video/mp4\">\n",
        "  </video>\n",
        "  \"\"\")\n",
        "\n",
        "# Get the CPU version and store working \n",
        "#!git clone https://github.com/NathanUA/U-2-Net.git\n",
        "![ ! -d /content/U-2-Net ] && git clone https://github.com/MrPalais/U-2-Net_Z.git && mv U-2-Net_Z /content/U-2-Net\n",
        "#![ -d /content/drive/MyDrive/U-2-Net ] && ln -f -s /content/drive/MyDrive/U-2-Net U-2-Net || echo \"ERROR with symbolic link! $? $!\" \n",
        "!ls -l /content\n",
        "\n",
        "# youtube downloader binary\n",
        "![ ! -f /content/U-2-Net/youtube-dl ] && curl -L https://yt-dl.org/downloads/latest/youtube-dl -o /content/U-2-Net/youtube-dl && chmod +x /content/U-2-Net/youtube-dl\n",
        "\n",
        "# Models\n",
        "!mkdir -p /content/U-2-Net/saved_models/u2net_portrait\n",
        "%cd /content/U-2-Net/saved_models/u2net_portrait\n",
        "! [ ! -f /content/U-2-Net/saved_models/u2net_portrait/u2net_portrait.pth ] && gdown --id 1IG3HdpcRiDoWNookbncQjeaPN28t90yW\n",
        "%cd /content/U-2-Net\n",
        "#clear_output()\n",
        "if len(os.listdir('/content/U-2-Net/saved_models/u2net_portrait')) > 0: print('OK!')\n",
        "else: print('Init ERROR pls check models.')\n",
        "\n",
        "\n"
      ],
      "execution_count": 1,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "Mounted at /content/drive\n",
            "Fri Aug 20 08:13:53 2021       \n",
            "+-----------------------------------------------------------------------------+\n",
            "| NVIDIA-SMI 470.57.02    Driver Version: 460.32.03    CUDA Version: 11.2     |\n",
            "|-------------------------------+----------------------+----------------------+\n",
            "| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |\n",
            "| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |\n",
            "|                               |                      |               MIG M. |\n",
            "|===============================+======================+======================|\n",
            "|   0  Tesla K80           Off  | 00000000:00:04.0 Off |                    0 |\n",
            "| N/A   37C    P8    26W / 149W |      0MiB / 11441MiB |      0%      Default |\n",
            "|                               |                      |                  N/A |\n",
            "+-------------------------------+----------------------+----------------------+\n",
            "                                                                               \n",
            "+-----------------------------------------------------------------------------+\n",
            "| Processes:                                                                  |\n",
            "|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |\n",
            "|        ID   ID                                                   Usage      |\n",
            "|=============================================================================|\n",
            "|  No running processes found                                                 |\n",
            "+-----------------------------------------------------------------------------+\n",
            "Collecting ffmpeg\n",
            "  Downloading ffmpeg-1.4.tar.gz (5.1 kB)\n",
            "Building wheels for collected packages: ffmpeg\n",
            "  Building wheel for ffmpeg (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
            "  Created wheel for ffmpeg: filename=ffmpeg-1.4-py3-none-any.whl size=6083 sha256=2aea8b691d1e200c44c5d84e5fb3eabffe943eb5eabe74a9620af9e1c6ecbef6\n",
            "  Stored in directory: /root/.cache/pip/wheels/64/80/6e/caa3e16deb0267c3cbfd36862058a724144e19fdb9eb03af0f\n",
            "Successfully built ffmpeg\n",
            "Installing collected packages: ffmpeg\n",
            "Successfully installed ffmpeg-1.4\n",
            "Collecting youtube-dl\n",
            "  Downloading youtube_dl-2021.6.6-py2.py3-none-any.whl (1.9 MB)\n",
            "\u001b[K     |████████████████████████████████| 1.9 MB 4.2 MB/s \n",
            "\u001b[?25hInstalling collected packages: youtube-dl\n",
            "Successfully installed youtube-dl-2021.6.6\n",
            "Imageio: 'ffmpeg-linux64-v3.3.1' was not found on your computer; downloading it now.\n",
            "Try 1. Download from https://github.com/imageio/imageio-binaries/raw/master/ffmpeg/ffmpeg-linux64-v3.3.1 (43.8 MB)\n",
            "Downloading: 8192/45929032 bytes (0.0%)\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b2088960/45929032 bytes (4.5%)\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b5505024/45929032 bytes (12.0%)\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b9003008/45929032 bytes (19.6%)\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b12083200/45929032 bytes (26.3%)\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b15458304/45929032 bytes (33.7%)\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b18898944/45929032 bytes (41.1%)\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b22355968/45929032 bytes (48.7%)\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b25657344/45929032 bytes (55.9%)\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b29081600/45929032 bytes (63.3%)\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b32522240/45929032 bytes (70.8%)\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b35815424/45929032 bytes (78.0%)\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b39206912/45929032 bytes (85.4%)\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b42598400/45929032 bytes (92.7%)\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b45929032/45929032 bytes (100.0%)\n",
            "  Done\n",
            "File saved as /root/.imageio/ffmpeg/ffmpeg-linux64-v3.3.1.\n",
            "Cloning into 'U-2-Net_Z'...\n",
            "remote: Enumerating objects: 853, done.\u001b[K\n",
            "remote: Counting objects: 100% (853/853), done.\u001b[K\n",
            "remote: Compressing objects: 100% (457/457), done.\u001b[K\n",
            "remote: Total 853 (delta 414), reused 815 (delta 392), pack-reused 0\u001b[K\n",
            "Receiving objects: 100% (853/853), 30.18 MiB | 12.83 MiB/s, done.\n",
            "Resolving deltas: 100% (414/414), done.\n",
            "total 12\n",
            "drwx------ 5 root root 4096 Aug 20 08:13 drive\n",
            "drwxr-xr-x 1 root root 4096 Aug 13 13:35 sample_data\n",
            "drwxr-xr-x 9 root root 4096 Aug 20 08:14 U-2-Net\n",
            "  % Total    % Received % Xferd  Average Speed   Time    Time     Time  Current\n",
            "                                 Dload  Upload   Total   Spent    Left  Speed\n",
            "100     3  100     3    0     0      1      0  0:00:03  0:00:02  0:00:01     1\n",
            "100     3  100     3    0     0      1      0  0:00:03  0:00:02  0:00:01     1\n",
            "100   614  100   614    0     0    192      0  0:00:03  0:00:03 --:--:--  599k\n",
            "100 1797k  100 1797k    0     0   496k      0  0:00:03  0:00:03 --:--:--  496k\n",
            "/content/U-2-Net/saved_models/u2net_portrait\n",
            "Downloading...\n",
            "From: https://drive.google.com/uc?id=1IG3HdpcRiDoWNookbncQjeaPN28t90yW\n",
            "To: /content/U-2-Net/saved_models/u2net_portrait/u2net_portrait.pth\n",
            "176MB [00:01, 89.3MB/s]\n",
            "/content/U-2-Net\n",
            "OK!\n"
          ],
          "name": "stdout"
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "3LI4ith2BKZ7",
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "outputId": "075a9422-e736-4e9d-f89d-627369444a7f"
      },
      "source": [
        "#@title <b><font color=\"red\" size=\"+3\">←</font><font color=\"black\" size=\"+3\">Get image/video</font></b>\n",
        "content = 'video' #@param [\"photo\", \"video\"]\n",
        "#@markdown *Ниже введите ссылку на фото или видео (например, YouTube), либо оставьте поле **source_url** пустым (в таком случае будет предложено загрузить файл с компьютера).*\n",
        "source_url = '/content/drive/MyDrive/VIDEO/chouquette.mp4' #@param {type:\"string\"}\n",
        "#@markdown *Откуда бы вы не грузили видео к нему будут применены следующие границы обрезки.*\n",
        "source_start = '00:00:35' #@param {type:\"string\"}\n",
        "source_end = '00:00:50' #@param {type:\"string\"}\n",
        "#@markdown *Если хотите оставить первоначальную длинну видео, то полностью сотрите эти две верхние строки.*\n",
        "from IPython.display import Image, display\n",
        "\n",
        "!rm -rf /content/U-2-Net/test_data/test_portrait_images/your_portrait_im/*.*\n",
        "if content == \"video\":\n",
        "  !rm -rf /content/source.mp4\n",
        "  if source_url == '':\n",
        "    uploaded = files.upload()\n",
        "    for fn in uploaded.keys():\n",
        "      print('User uploaded file \"{name}\" with length {length} bytes'.format(\n",
        "          name=fn, length=len(uploaded[fn])))\n",
        "    if source_start!='' or source_end!='': \n",
        "      !ffmpeg -y -i $fn -ss $source_start -to $source_end -r 25 /content/source.mp4\n",
        "    else: \n",
        "      !ffmpeg -y -i $fn -r 25 /content/source.mp4\n",
        "    !rm $fn\n",
        "    file_name = '/content/source.mp4'\n",
        "  else:\n",
        "    if 'youtu' in source_url:\n",
        "      !rm -rf /content/source*\n",
        "      !/content/U-2-Net/youtube-dl $source_url -f bestvideo+bestaudio --merge-output-format mkv -o /content/source_tmp.mkv\n",
        "      if source_start!='' or source_end!='': \n",
        "        !ffmpeg -y -i /content/source_tmp.mkv -ss $source_start -to $source_end -r 25 /content/source.mp4\n",
        "      else: \n",
        "        !ffmpeg -y -i /content/source_tmp.mkv -r 25 /content/source.mp4\n",
        "      !rm /content/source_tmp.mkv\n",
        "      file_name = '/content/source.mp4'\n",
        "    else:\n",
        "      fn = source_url.split('/')[-1]\n",
        "      !wget $source_url || cp $source_url $fn\n",
        "      !ls -l    \n",
        "      print (\"Converting \"+fn+\" to 25fps\")\n",
        "      if source_start!='' or source_end!='': \n",
        "        !ffmpeg -y -i $fn -ss $source_start -to $source_end -r 25 -crf 20 -vf scale=-1:720 /content/source.mp4\n",
        "      else: \n",
        "        #!ffmpeg -y -i $fn -r 25 /content/source.mp4\n",
        "        !echo \"Assuming that the frame rate is 25FPS...\"\n",
        "        !cp $fn /content/source.mp4\n",
        "      !rm $fn\n",
        "      file_name = '/content/source.mp4'\n",
        "  try: \n",
        "    #clear_output()\n",
        "    print('generation of pictures.')\n",
        "    vidcap = cv2.VideoCapture(file_name)\n",
        "    success,image = vidcap.read()\n",
        "    count = 0\n",
        "    success = True\n",
        "    while success:\n",
        "      cv2.imwrite(\"/content/U-2-Net/test_data/test_portrait_images/your_portrait_im/%09d.png\" % count, image)\n",
        "      success,image = vidcap.read()\n",
        "      count += 1\n",
        "    fps_of_video = int(cv2.VideoCapture(file_name).get(cv2.CAP_PROP_FPS))\n",
        "    frames_of_video = int(cv2.VideoCapture(file_name).get(cv2.CAP_PROP_FRAME_COUNT))\n",
        "    #clear_output()\n",
        "    print('Видео разбито на {} кадров'.format(str(frames_of_video)))\n",
        "  except: \n",
        "    #clear_output()\n",
        "    print(\"Error\")\n",
        "else:\n",
        "  if source_url == '':\n",
        "    uploaded = files.upload()\n",
        "    for fn in uploaded.keys():\n",
        "      print('User uploaded file \"{name}\" with length {length} bytes'.format(\n",
        "          name=fn, length=len(uploaded[fn])))\n",
        "    os.rename(fn, fn.replace(\" \", \"\"))\n",
        "    fn = fn.replace(\" \", \"\")\n",
        "    pic_name = \"/content/U-2-Net/test_data/test_portrait_images/your_portrait_im/photo.\" + fn.split(\".\")[-1]\n",
        "    !mv -f $fn $pic_name\n",
        "    #clear_output()\n",
        "    print('Фото успешно загружено')\n",
        "  else:\n",
        "    try:\n",
        "      pic_name = \"/content/U-2-Net/test_data/test_portrait_images/your_portrait_im/photo.\" + source_url.split('.')[-1]\n",
        "      !wget $source_url -O $pic_name\n",
        "      #clear_output()\n",
        "      print('Фото успешно загружено')\n",
        "    except:\n",
        "      #clear_output() \n",
        "      print('Что-то пошло не так. Попробуйте загрузить фото с компьютера.')"
      ],
      "execution_count": 13,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "/content/drive/MyDrive/VIDEO/chouquette.mp4: Scheme missing.\n",
            "total 208412\n",
            "-rw------- 1 root root 211216338 Aug 20 08:52  chouquette.mp4\n",
            "-rw-r--r-- 1 root root    206226 Aug 20 08:14  clipping_camera.jpg\n",
            "-rw-r--r-- 1 root root      9040 Aug 20 08:14  data_loader.py\n",
            "drwxr-xr-x 2 root root      4096 Aug 20 08:14  figures\n",
            "drwxr-xr-x 2 root root      4096 Aug 20 08:14  gradio\n",
            "-rw-r--r-- 1 root root     11357 Aug 20 08:14  LICENSE\n",
            "drwxr-xr-x 3 root root      4096 Aug 20 08:14  model\n",
            "-rw-r--r-- 1 root root     34237 Aug 20 08:14 'Portrait_Painter_Z_(CPU_or_GPU).ipynb'\n",
            "drwxr-xr-x 2 root root      4096 Aug 20 08:14  __pycache__\n",
            "-rw-r--r-- 1 root root     12886 Aug 20 08:14  README.md\n",
            "-rw-r--r-- 1 root root       111 Aug 20 08:14  requirements.txt\n",
            "drwxr-xr-x 4 root root      4096 Aug 20 08:14  saved_models\n",
            "-rw-r--r-- 1 root root       431 Aug 20 08:14  setup_model_weights.py\n",
            "drwxr-xr-x 8 root root      4096 Aug 20 08:14  test_data\n",
            "-rw-r--r-- 1 root root      3537 Aug 20 08:14  u2net_human_seg_test.py\n",
            "-rw-r--r-- 1 root root      4970 Aug 20 08:14  u2net_portrait_demo.py\n",
            "-rw-r--r-- 1 root root      3399 Aug 20 08:14  u2net_portrait_test.py\n",
            "-rw-r--r-- 1 root root      3659 Aug 20 08:14  u2net_test.py\n",
            "-rw-r--r-- 1 root root      4995 Aug 20 08:14  u2net_train.py\n",
            "-rwxr-xr-x 1 root root   1840311 Aug 20 08:14  youtube-dl\n",
            "Converting chouquette.mp4 to 25fps\n",
            "ffmpeg version 3.4.8-0ubuntu0.2 Copyright (c) 2000-2020 the FFmpeg developers\n",
            "  built with gcc 7 (Ubuntu 7.5.0-3ubuntu1~18.04)\n",
            "  configuration: --prefix=/usr --extra-version=0ubuntu0.2 --toolchain=hardened --libdir=/usr/lib/x86_64-linux-gnu --incdir=/usr/include/x86_64-linux-gnu --enable-gpl --disable-stripping --enable-avresample --enable-avisynth --enable-gnutls --enable-ladspa --enable-libass --enable-libbluray --enable-libbs2b --enable-libcaca --enable-libcdio --enable-libflite --enable-libfontconfig --enable-libfreetype --enable-libfribidi --enable-libgme --enable-libgsm --enable-libmp3lame --enable-libmysofa --enable-libopenjpeg --enable-libopenmpt --enable-libopus --enable-libpulse --enable-librubberband --enable-librsvg --enable-libshine --enable-libsnappy --enable-libsoxr --enable-libspeex --enable-libssh --enable-libtheora --enable-libtwolame --enable-libvorbis --enable-libvpx --enable-libwavpack --enable-libwebp --enable-libx265 --enable-libxml2 --enable-libxvid --enable-libzmq --enable-libzvbi --enable-omx --enable-openal --enable-opengl --enable-sdl2 --enable-libdc1394 --enable-libdrm --enable-libiec61883 --enable-chromaprint --enable-frei0r --enable-libopencv --enable-libx264 --enable-shared\n",
            "  libavutil      55. 78.100 / 55. 78.100\n",
            "  libavcodec     57.107.100 / 57.107.100\n",
            "  libavformat    57. 83.100 / 57. 83.100\n",
            "  libavdevice    57. 10.100 / 57. 10.100\n",
            "  libavfilter     6.107.100 /  6.107.100\n",
            "  libavresample   3.  7.  0 /  3.  7.  0\n",
            "  libswscale      4.  8.100 /  4.  8.100\n",
            "  libswresample   2.  9.100 /  2.  9.100\n",
            "  libpostproc    54.  7.100 / 54.  7.100\n",
            "Input #0, mov,mp4,m4a,3gp,3g2,mj2, from 'chouquette.mp4':\n",
            "  Metadata:\n",
            "    major_brand     : mp42\n",
            "    minor_version   : 0\n",
            "    compatible_brands: isommp42\n",
            "    creation_time   : 2021-07-14T19:05:39.000000Z\n",
            "  Duration: 00:02:27.60, start: 0.000000, bitrate: 11448 kb/s\n",
            "    Stream #0:0(eng): Video: hevc (Main) (hvc1 / 0x31637668), yuv420p(tv), 1920x1080 [SAR 1:1 DAR 16:9], 11255 kb/s, 25 fps, 25 tbr, 25k tbn, 25 tbc (default)\n",
            "    Metadata:\n",
            "      creation_time   : 2021-07-14T19:05:41.000000Z\n",
            "      handler_name    : Mainconcept MP4 Video Media Handler\n",
            "      encoder         : HEVC Coding\n",
            "    Stream #0:1(eng): Audio: aac (LC) (mp4a / 0x6134706D), 48000 Hz, stereo, fltp, 189 kb/s (default)\n",
            "    Metadata:\n",
            "      creation_time   : 2021-07-14T19:05:41.000000Z\n",
            "      handler_name    : Mainconcept MP4 Sound Media Handler\n",
            "Stream mapping:\n",
            "  Stream #0:0 -> #0:0 (hevc (native) -> h264 (libx264))\n",
            "  Stream #0:1 -> #0:1 (aac (native) -> aac (native))\n",
            "Press [q] to stop, [?] for help\n",
            "\u001b[1;36m[libx264 @ 0x55b9d53f0800] \u001b[0musing SAR=32/27\n",
            "\u001b[1;36m[libx264 @ 0x55b9d53f0800] \u001b[0musing cpu capabilities: MMX2 SSE2Fast SSSE3 SSE4.2 AVX FMA3 BMI2 AVX2\n",
            "\u001b[1;36m[libx264 @ 0x55b9d53f0800] \u001b[0mprofile High, level 3.1\n",
            "\u001b[1;36m[libx264 @ 0x55b9d53f0800] \u001b[0m264 - core 152 r2854 e9a5903 - H.264/MPEG-4 AVC codec - Copyleft 2003-2017 - http://www.videolan.org/x264.html - options: cabac=1 ref=3 deblock=1:0:0 analyse=0x3:0x113 me=hex subme=7 psy=1 psy_rd=1.00:0.00 mixed_ref=1 me_range=16 chroma_me=1 trellis=1 8x8dct=1 cqm=0 deadzone=21,11 fast_pskip=1 chroma_qp_offset=-2 threads=3 lookahead_threads=1 sliced_threads=0 nr=0 decimate=1 interlaced=0 bluray_compat=0 constrained_intra=0 bframes=3 b_pyramid=2 b_adapt=1 b_bias=0 direct=1 weightb=1 open_gop=0 weightp=2 keyint=250 keyint_min=25 scenecut=40 intra_refresh=0 rc_lookahead=40 rc=crf mbtree=1 crf=23.0 qcomp=0.60 qpmin=0 qpmax=69 qpstep=4 ip_ratio=1.40 aq=1:1.00\n",
            "Output #0, mp4, to '/content/source.mp4':\n",
            "  Metadata:\n",
            "    major_brand     : mp42\n",
            "    minor_version   : 0\n",
            "    compatible_brands: isommp42\n",
            "    encoder         : Lavf57.83.100\n",
            "    Stream #0:0(eng): Video: h264 (libx264) (avc1 / 0x31637661), yuv420p, 1080x720 [SAR 32:27 DAR 16:9], q=-1--1, 25 fps, 12800 tbn, 25 tbc (default)\n",
            "    Metadata:\n",
            "      creation_time   : 2021-07-14T19:05:41.000000Z\n",
            "      handler_name    : Mainconcept MP4 Video Media Handler\n",
            "      encoder         : Lavc57.107.100 libx264\n",
            "    Side data:\n",
            "      cpb: bitrate max/min/avg: 0/0/0 buffer size: 0 vbv_delay: -1\n",
            "    Stream #0:1(eng): Audio: aac (LC) (mp4a / 0x6134706D), 48000 Hz, stereo, fltp, 128 kb/s (default)\n",
            "    Metadata:\n",
            "      creation_time   : 2021-07-14T19:05:41.000000Z\n",
            "      handler_name    : Mainconcept MP4 Sound Media Handler\n",
            "      encoder         : Lavc57.107.100 aac\n",
            "frame=  375 fps=8.6 q=-1.0 Lsize=    3417kB time=00:00:15.01 bitrate=1863.8kbits/s speed=0.345x    \n",
            "video:3151kB audio:254kB subtitle:0kB other streams:0kB global headers:0kB muxing overhead: 0.366570%\n",
            "\u001b[1;36m[libx264 @ 0x55b9d53f0800] \u001b[0mframe I:3     Avg QP:20.86  size: 14148\n",
            "\u001b[1;36m[libx264 @ 0x55b9d53f0800] \u001b[0mframe P:100   Avg QP:27.22  size: 16612\n",
            "\u001b[1;36m[libx264 @ 0x55b9d53f0800] \u001b[0mframe B:272   Avg QP:28.82  size:  5596\n",
            "\u001b[1;36m[libx264 @ 0x55b9d53f0800] \u001b[0mconsecutive B-frames:  1.6%  2.7%  7.2% 88.5%\n",
            "\u001b[1;36m[libx264 @ 0x55b9d53f0800] \u001b[0mmb I  I16..4: 20.8% 74.5%  4.7%\n",
            "\u001b[1;36m[libx264 @ 0x55b9d53f0800] \u001b[0mmb P  I16..4:  0.4%  9.4%  1.6%  P16..4: 11.8%  9.6%  5.2%  0.0%  0.0%    skip:62.0%\n",
            "\u001b[1;36m[libx264 @ 0x55b9d53f0800] \u001b[0mmb B  I16..4:  0.0%  1.1%  0.1%  B16..8: 19.9%  6.4%  1.6%  direct: 2.3%  skip:68.6%  L0:39.8% L1:41.7% BI:18.5%\n",
            "\u001b[1;36m[libx264 @ 0x55b9d53f0800] \u001b[0m8x8 transform intra:82.6% inter:74.8%\n",
            "\u001b[1;36m[libx264 @ 0x55b9d53f0800] \u001b[0mcoded y,uvDC,uvAC intra: 64.7% 61.7% 27.3% inter: 14.2% 13.2% 1.0%\n",
            "\u001b[1;36m[libx264 @ 0x55b9d53f0800] \u001b[0mi16 v,h,dc,p: 70%  2%  7% 21%\n",
            "\u001b[1;36m[libx264 @ 0x55b9d53f0800] \u001b[0mi8 v,h,dc,ddl,ddr,vr,hd,vl,hu: 22%  5% 27%  6%  7% 12%  4% 11%  5%\n",
            "\u001b[1;36m[libx264 @ 0x55b9d53f0800] \u001b[0mi4 v,h,dc,ddl,ddr,vr,hd,vl,hu: 25%  9% 14%  8% 10% 14%  6% 10%  5%\n",
            "\u001b[1;36m[libx264 @ 0x55b9d53f0800] \u001b[0mi8c dc,h,v,p: 46%  7% 28% 19%\n",
            "\u001b[1;36m[libx264 @ 0x55b9d53f0800] \u001b[0mWeighted P-Frames: Y:8.0% UV:3.0%\n",
            "\u001b[1;36m[libx264 @ 0x55b9d53f0800] \u001b[0mref P L0: 58.2% 20.3% 15.5%  5.5%  0.4%\n",
            "\u001b[1;36m[libx264 @ 0x55b9d53f0800] \u001b[0mref B L0: 94.3%  4.7%  0.9%\n",
            "\u001b[1;36m[libx264 @ 0x55b9d53f0800] \u001b[0mref B L1: 98.3%  1.7%\n",
            "\u001b[1;36m[libx264 @ 0x55b9d53f0800] \u001b[0mkb/s:1720.43\n",
            "\u001b[1;36m[aac @ 0x55b9d53f1700] \u001b[0mQavg: 755.172\n",
            "generation of pictures.\n",
            "Видео разбито на 375 кадров\n"
          ],
          "name": "stdout"
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "n5QboJhJTOBf",
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "outputId": "57798516-5b49-4880-fb86-04c1d41c4ce3"
      },
      "source": [
        "#@title <b><font color=\"red\" size=\"+3\">←</font><font color=\"black\" size=\"+3\"> Начинаем отрисовку</font></b>\n",
        "\n",
        "!rm -rf /content/U-2-Net/test_data/test_portrait_images/your_portrait_results/*.*\n",
        "!python u2net_portrait_demo.py\n",
        "#clear_output()\n",
        "#!sleep 300\n",
        "if content == \"video\": \n",
        "  print('Generate video')\n",
        "  !rm -f /content/U-2-Net/result.mp4  \n",
        "  !ffmpeg -r $fps_of_video -i /content/U-2-Net/test_data/test_portrait_images/your_portrait_results/%09d.png -c:v prores -an /content/U-2-Net/result.mp4\n",
        "  path_to_res = '/content/U-2-Net/result.mp4'\n",
        "  #clear_output()\n",
        "  print('ready')\n",
        "else:\n",
        "  path_to_imgs = '/content/U-2-Net/test_data/test_portrait_images/your_portrait_results/'\n",
        "  list_pic = os.listdir(path_to_imgs)\n",
        "  if len(list_pic) == 1:\n",
        "    path_to_res = path_to_imgs + list_pic[0]\n",
        "    clear_output()\n",
        "    cv2_imshow(cv2.imread(path_to_res))\n",
        "  elif len(list_pic) > 1:\n",
        "    %cd $path_to_imgs\n",
        "    !zip archive *\n",
        "    %cd /content/U-2-Net\n",
        "    path_to_res = path_to_imgs+'archive.zip'\n",
        "    #clear_output()\n",
        "    print('Архив с картинками сформирован')\n",
        "  else: print('Что-то пошло не так')"
      ],
      "execution_count": 14,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "Number of images:  375\n",
            "FAST CUDA MODE ACTIVATED!\n",
            "--------------------------\n",
            "inferencing  0 / 375 ./test_data/test_portrait_images/your_portrait_im/000000285.png\n",
            "/usr/local/lib/python3.7/dist-packages/torch/nn/functional.py:718: UserWarning: Named tensors and all their associated APIs are an experimental feature and subject to change. Please do not use them for anything important until they are released as stable. (Triggered internally at  /pytorch/c10/core/TensorImpl.h:1156.)\n",
            "  return torch.max_pool2d(input, kernel_size, stride, padding, dilation, ceil_mode)\n",
            "/usr/local/lib/python3.7/dist-packages/torch/nn/functional.py:3487: UserWarning: nn.functional.upsample is deprecated. Use nn.functional.interpolate instead.\n",
            "  warnings.warn(\"nn.functional.upsample is deprecated. Use nn.functional.interpolate instead.\")\n",
            "/usr/local/lib/python3.7/dist-packages/torch/nn/functional.py:3613: UserWarning: Default upsampling behavior when mode=bilinear is changed to align_corners=False since 0.4.0. Please specify align_corners=True if the old behavior is desired. See the documentation of nn.Upsample for details.\n",
            "  \"See the documentation of nn.Upsample for details.\".format(mode)\n",
            "/usr/local/lib/python3.7/dist-packages/torch/nn/functional.py:1805: UserWarning: nn.functional.sigmoid is deprecated. Use torch.sigmoid instead.\n",
            "  warnings.warn(\"nn.functional.sigmoid is deprecated. Use torch.sigmoid instead.\")\n",
            "--------------------------\n",
            "inferencing  1 / 375 ./test_data/test_portrait_images/your_portrait_im/000000153.png\n",
            "--------------------------\n",
            "inferencing  2 / 375 ./test_data/test_portrait_images/your_portrait_im/000000144.png\n",
            "--------------------------\n",
            "inferencing  3 / 375 ./test_data/test_portrait_images/your_portrait_im/000000266.png\n",
            "--------------------------\n",
            "inferencing  4 / 375 ./test_data/test_portrait_images/your_portrait_im/000000009.png\n",
            "--------------------------\n",
            "inferencing  5 / 375 ./test_data/test_portrait_images/your_portrait_im/000000157.png\n",
            "--------------------------\n",
            "inferencing  6 / 375 ./test_data/test_portrait_images/your_portrait_im/000000017.png\n",
            "--------------------------\n",
            "inferencing  7 / 375 ./test_data/test_portrait_images/your_portrait_im/000000014.png\n",
            "--------------------------\n",
            "inferencing  8 / 375 ./test_data/test_portrait_images/your_portrait_im/000000116.png\n",
            "--------------------------\n",
            "inferencing  9 / 375 ./test_data/test_portrait_images/your_portrait_im/000000353.png\n",
            "--------------------------\n",
            "inferencing  10 / 375 ./test_data/test_portrait_images/your_portrait_im/000000218.png\n",
            "--------------------------\n",
            "inferencing  11 / 375 ./test_data/test_portrait_images/your_portrait_im/000000164.png\n",
            "--------------------------\n",
            "inferencing  12 / 375 ./test_data/test_portrait_images/your_portrait_im/000000008.png\n",
            "--------------------------\n",
            "inferencing  13 / 375 ./test_data/test_portrait_images/your_portrait_im/000000097.png\n",
            "--------------------------\n",
            "inferencing  14 / 375 ./test_data/test_portrait_images/your_portrait_im/000000339.png\n",
            "--------------------------\n",
            "inferencing  15 / 375 ./test_data/test_portrait_images/your_portrait_im/000000246.png\n",
            "--------------------------\n",
            "inferencing  16 / 375 ./test_data/test_portrait_images/your_portrait_im/000000089.png\n",
            "--------------------------\n",
            "inferencing  17 / 375 ./test_data/test_portrait_images/your_portrait_im/000000265.png\n",
            "--------------------------\n",
            "inferencing  18 / 375 ./test_data/test_portrait_images/your_portrait_im/000000107.png\n",
            "--------------------------\n",
            "inferencing  19 / 375 ./test_data/test_portrait_images/your_portrait_im/000000038.png\n",
            "--------------------------\n",
            "inferencing  20 / 375 ./test_data/test_portrait_images/your_portrait_im/000000175.png\n",
            "--------------------------\n",
            "inferencing  21 / 375 ./test_data/test_portrait_images/your_portrait_im/000000244.png\n",
            "--------------------------\n",
            "inferencing  22 / 375 ./test_data/test_portrait_images/your_portrait_im/000000085.png\n",
            "--------------------------\n",
            "inferencing  23 / 375 ./test_data/test_portrait_images/your_portrait_im/000000293.png\n",
            "--------------------------\n",
            "inferencing  24 / 375 ./test_data/test_portrait_images/your_portrait_im/000000249.png\n",
            "--------------------------\n",
            "inferencing  25 / 375 ./test_data/test_portrait_images/your_portrait_im/000000203.png\n",
            "--------------------------\n",
            "inferencing  26 / 375 ./test_data/test_portrait_images/your_portrait_im/000000019.png\n",
            "--------------------------\n",
            "inferencing  27 / 375 ./test_data/test_portrait_images/your_portrait_im/000000247.png\n",
            "--------------------------\n",
            "inferencing  28 / 375 ./test_data/test_portrait_images/your_portrait_im/000000074.png\n",
            "--------------------------\n",
            "inferencing  29 / 375 ./test_data/test_portrait_images/your_portrait_im/000000359.png\n",
            "--------------------------\n",
            "inferencing  30 / 375 ./test_data/test_portrait_images/your_portrait_im/000000062.png\n",
            "--------------------------\n",
            "inferencing  31 / 375 ./test_data/test_portrait_images/your_portrait_im/000000214.png\n",
            "--------------------------\n",
            "inferencing  32 / 375 ./test_data/test_portrait_images/your_portrait_im/000000252.png\n",
            "--------------------------\n",
            "inferencing  33 / 375 ./test_data/test_portrait_images/your_portrait_im/000000337.png\n",
            "--------------------------\n",
            "inferencing  34 / 375 ./test_data/test_portrait_images/your_portrait_im/000000330.png\n",
            "--------------------------\n",
            "inferencing  35 / 375 ./test_data/test_portrait_images/your_portrait_im/000000037.png\n",
            "--------------------------\n",
            "inferencing  36 / 375 ./test_data/test_portrait_images/your_portrait_im/000000277.png\n",
            "--------------------------\n",
            "inferencing  37 / 375 ./test_data/test_portrait_images/your_portrait_im/000000177.png\n",
            "--------------------------\n",
            "inferencing  38 / 375 ./test_data/test_portrait_images/your_portrait_im/000000069.png\n",
            "--------------------------\n",
            "inferencing  39 / 375 ./test_data/test_portrait_images/your_portrait_im/000000207.png\n",
            "--------------------------\n",
            "inferencing  40 / 375 ./test_data/test_portrait_images/your_portrait_im/000000184.png\n",
            "--------------------------\n",
            "inferencing  41 / 375 ./test_data/test_portrait_images/your_portrait_im/000000142.png\n",
            "--------------------------\n",
            "inferencing  42 / 375 ./test_data/test_portrait_images/your_portrait_im/000000106.png\n",
            "--------------------------\n",
            "inferencing  43 / 375 ./test_data/test_portrait_images/your_portrait_im/000000232.png\n",
            "--------------------------\n",
            "inferencing  44 / 375 ./test_data/test_portrait_images/your_portrait_im/000000273.png\n",
            "--------------------------\n",
            "inferencing  45 / 375 ./test_data/test_portrait_images/your_portrait_im/000000007.png\n",
            "--------------------------\n",
            "inferencing  46 / 375 ./test_data/test_portrait_images/your_portrait_im/000000171.png\n",
            "--------------------------\n",
            "inferencing  47 / 375 ./test_data/test_portrait_images/your_portrait_im/000000282.png\n",
            "--------------------------\n",
            "inferencing  48 / 375 ./test_data/test_portrait_images/your_portrait_im/000000044.png\n",
            "--------------------------\n",
            "inferencing  49 / 375 ./test_data/test_portrait_images/your_portrait_im/000000366.png\n",
            "--------------------------\n",
            "inferencing  50 / 375 ./test_data/test_portrait_images/your_portrait_im/000000172.png\n",
            "--------------------------\n",
            "inferencing  51 / 375 ./test_data/test_portrait_images/your_portrait_im/000000358.png\n",
            "--------------------------\n",
            "inferencing  52 / 375 ./test_data/test_portrait_images/your_portrait_im/000000105.png\n",
            "--------------------------\n",
            "inferencing  53 / 375 ./test_data/test_portrait_images/your_portrait_im/000000314.png\n",
            "--------------------------\n",
            "inferencing  54 / 375 ./test_data/test_portrait_images/your_portrait_im/000000197.png\n",
            "--------------------------\n",
            "inferencing  55 / 375 ./test_data/test_portrait_images/your_portrait_im/000000242.png\n",
            "--------------------------\n",
            "inferencing  56 / 375 ./test_data/test_portrait_images/your_portrait_im/000000317.png\n",
            "--------------------------\n",
            "inferencing  57 / 375 ./test_data/test_portrait_images/your_portrait_im/000000126.png\n",
            "--------------------------\n",
            "inferencing  58 / 375 ./test_data/test_portrait_images/your_portrait_im/000000195.png\n",
            "--------------------------\n",
            "inferencing  59 / 375 ./test_data/test_portrait_images/your_portrait_im/000000072.png\n",
            "--------------------------\n",
            "inferencing  60 / 375 ./test_data/test_portrait_images/your_portrait_im/000000290.png\n",
            "--------------------------\n",
            "inferencing  61 / 375 ./test_data/test_portrait_images/your_portrait_im/000000013.png\n",
            "--------------------------\n",
            "inferencing  62 / 375 ./test_data/test_portrait_images/your_portrait_im/000000345.png\n",
            "--------------------------\n",
            "inferencing  63 / 375 ./test_data/test_portrait_images/your_portrait_im/000000286.png\n",
            "--------------------------\n",
            "inferencing  64 / 375 ./test_data/test_portrait_images/your_portrait_im/000000312.png\n",
            "--------------------------\n",
            "inferencing  65 / 375 ./test_data/test_portrait_images/your_portrait_im/000000210.png\n",
            "--------------------------\n",
            "inferencing  66 / 375 ./test_data/test_portrait_images/your_portrait_im/000000258.png\n",
            "--------------------------\n",
            "inferencing  67 / 375 ./test_data/test_portrait_images/your_portrait_im/000000036.png\n",
            "--------------------------\n",
            "inferencing  68 / 375 ./test_data/test_portrait_images/your_portrait_im/000000040.png\n",
            "--------------------------\n",
            "inferencing  69 / 375 ./test_data/test_portrait_images/your_portrait_im/000000364.png\n",
            "--------------------------\n",
            "inferencing  70 / 375 ./test_data/test_portrait_images/your_portrait_im/000000225.png\n",
            "--------------------------\n",
            "inferencing  71 / 375 ./test_data/test_portrait_images/your_portrait_im/000000056.png\n",
            "--------------------------\n",
            "inferencing  72 / 375 ./test_data/test_portrait_images/your_portrait_im/000000256.png\n",
            "--------------------------\n",
            "inferencing  73 / 375 ./test_data/test_portrait_images/your_portrait_im/000000224.png\n",
            "--------------------------\n",
            "inferencing  74 / 375 ./test_data/test_portrait_images/your_portrait_im/000000117.png\n",
            "--------------------------\n",
            "inferencing  75 / 375 ./test_data/test_portrait_images/your_portrait_im/000000167.png\n",
            "--------------------------\n",
            "inferencing  76 / 375 ./test_data/test_portrait_images/your_portrait_im/000000163.png\n",
            "--------------------------\n",
            "inferencing  77 / 375 ./test_data/test_portrait_images/your_portrait_im/000000261.png\n",
            "--------------------------\n",
            "inferencing  78 / 375 ./test_data/test_portrait_images/your_portrait_im/000000289.png\n",
            "--------------------------\n",
            "inferencing  79 / 375 ./test_data/test_portrait_images/your_portrait_im/000000215.png\n",
            "--------------------------\n",
            "inferencing  80 / 375 ./test_data/test_portrait_images/your_portrait_im/000000100.png\n",
            "--------------------------\n",
            "inferencing  81 / 375 ./test_data/test_portrait_images/your_portrait_im/000000270.png\n",
            "--------------------------\n",
            "inferencing  82 / 375 ./test_data/test_portrait_images/your_portrait_im/000000076.png\n",
            "--------------------------\n",
            "inferencing  83 / 375 ./test_data/test_portrait_images/your_portrait_im/000000127.png\n",
            "--------------------------\n",
            "inferencing  84 / 375 ./test_data/test_portrait_images/your_portrait_im/000000006.png\n",
            "--------------------------\n",
            "inferencing  85 / 375 ./test_data/test_portrait_images/your_portrait_im/000000075.png\n",
            "--------------------------\n",
            "inferencing  86 / 375 ./test_data/test_portrait_images/your_portrait_im/000000154.png\n",
            "--------------------------\n",
            "inferencing  87 / 375 ./test_data/test_portrait_images/your_portrait_im/000000281.png\n",
            "--------------------------\n",
            "inferencing  88 / 375 ./test_data/test_portrait_images/your_portrait_im/000000299.png\n",
            "--------------------------\n",
            "inferencing  89 / 375 ./test_data/test_portrait_images/your_portrait_im/000000131.png\n",
            "--------------------------\n",
            "inferencing  90 / 375 ./test_data/test_portrait_images/your_portrait_im/000000279.png\n",
            "--------------------------\n",
            "inferencing  91 / 375 ./test_data/test_portrait_images/your_portrait_im/000000220.png\n",
            "--------------------------\n",
            "inferencing  92 / 375 ./test_data/test_portrait_images/your_portrait_im/000000287.png\n",
            "--------------------------\n",
            "inferencing  93 / 375 ./test_data/test_portrait_images/your_portrait_im/000000027.png\n",
            "--------------------------\n",
            "inferencing  94 / 375 ./test_data/test_portrait_images/your_portrait_im/000000068.png\n",
            "--------------------------\n",
            "inferencing  95 / 375 ./test_data/test_portrait_images/your_portrait_im/000000092.png\n",
            "--------------------------\n",
            "inferencing  96 / 375 ./test_data/test_portrait_images/your_portrait_im/000000005.png\n",
            "--------------------------\n",
            "inferencing  97 / 375 ./test_data/test_portrait_images/your_portrait_im/000000030.png\n",
            "--------------------------\n",
            "inferencing  98 / 375 ./test_data/test_portrait_images/your_portrait_im/000000070.png\n",
            "--------------------------\n",
            "inferencing  99 / 375 ./test_data/test_portrait_images/your_portrait_im/000000031.png\n",
            "--------------------------\n",
            "inferencing  100 / 375 ./test_data/test_portrait_images/your_portrait_im/000000352.png\n",
            "--------------------------\n",
            "inferencing  101 / 375 ./test_data/test_portrait_images/your_portrait_im/000000255.png\n",
            "--------------------------\n",
            "inferencing  102 / 375 ./test_data/test_portrait_images/your_portrait_im/000000010.png\n",
            "--------------------------\n",
            "inferencing  103 / 375 ./test_data/test_portrait_images/your_portrait_im/000000018.png\n",
            "--------------------------\n",
            "inferencing  104 / 375 ./test_data/test_portrait_images/your_portrait_im/000000088.png\n",
            "--------------------------\n",
            "inferencing  105 / 375 ./test_data/test_portrait_images/your_portrait_im/000000047.png\n",
            "--------------------------\n",
            "inferencing  106 / 375 ./test_data/test_portrait_images/your_portrait_im/000000189.png\n",
            "--------------------------\n",
            "inferencing  107 / 375 ./test_data/test_portrait_images/your_portrait_im/000000182.png\n",
            "--------------------------\n",
            "inferencing  108 / 375 ./test_data/test_portrait_images/your_portrait_im/000000003.png\n",
            "--------------------------\n",
            "inferencing  109 / 375 ./test_data/test_portrait_images/your_portrait_im/000000271.png\n",
            "--------------------------\n",
            "inferencing  110 / 375 ./test_data/test_portrait_images/your_portrait_im/000000041.png\n",
            "--------------------------\n",
            "inferencing  111 / 375 ./test_data/test_portrait_images/your_portrait_im/000000370.png\n",
            "--------------------------\n",
            "inferencing  112 / 375 ./test_data/test_portrait_images/your_portrait_im/000000004.png\n",
            "--------------------------\n",
            "inferencing  113 / 375 ./test_data/test_portrait_images/your_portrait_im/000000363.png\n",
            "--------------------------\n",
            "inferencing  114 / 375 ./test_data/test_portrait_images/your_portrait_im/000000025.png\n",
            "--------------------------\n",
            "inferencing  115 / 375 ./test_data/test_portrait_images/your_portrait_im/000000204.png\n",
            "--------------------------\n",
            "inferencing  116 / 375 ./test_data/test_portrait_images/your_portrait_im/000000124.png\n",
            "--------------------------\n",
            "inferencing  117 / 375 ./test_data/test_portrait_images/your_portrait_im/000000374.png\n",
            "--------------------------\n",
            "inferencing  118 / 375 ./test_data/test_portrait_images/your_portrait_im/000000313.png\n",
            "--------------------------\n",
            "inferencing  119 / 375 ./test_data/test_portrait_images/your_portrait_im/000000316.png\n",
            "--------------------------\n",
            "inferencing  120 / 375 ./test_data/test_portrait_images/your_portrait_im/000000326.png\n",
            "--------------------------\n",
            "inferencing  121 / 375 ./test_data/test_portrait_images/your_portrait_im/000000122.png\n",
            "--------------------------\n",
            "inferencing  122 / 375 ./test_data/test_portrait_images/your_portrait_im/000000294.png\n",
            "--------------------------\n",
            "inferencing  123 / 375 ./test_data/test_portrait_images/your_portrait_im/000000086.png\n",
            "--------------------------\n",
            "inferencing  124 / 375 ./test_data/test_portrait_images/your_portrait_im/000000015.png\n",
            "--------------------------\n",
            "inferencing  125 / 375 ./test_data/test_portrait_images/your_portrait_im/000000348.png\n",
            "--------------------------\n",
            "inferencing  126 / 375 ./test_data/test_portrait_images/your_portrait_im/000000309.png\n",
            "--------------------------\n",
            "inferencing  127 / 375 ./test_data/test_portrait_images/your_portrait_im/000000140.png\n",
            "--------------------------\n",
            "inferencing  128 / 375 ./test_data/test_portrait_images/your_portrait_im/000000045.png\n",
            "--------------------------\n",
            "inferencing  129 / 375 ./test_data/test_portrait_images/your_portrait_im/000000091.png\n",
            "--------------------------\n",
            "inferencing  130 / 375 ./test_data/test_portrait_images/your_portrait_im/000000229.png\n",
            "--------------------------\n",
            "inferencing  131 / 375 ./test_data/test_portrait_images/your_portrait_im/000000371.png\n",
            "--------------------------\n",
            "inferencing  132 / 375 ./test_data/test_portrait_images/your_portrait_im/000000221.png\n",
            "--------------------------\n",
            "inferencing  133 / 375 ./test_data/test_portrait_images/your_portrait_im/000000357.png\n",
            "--------------------------\n",
            "inferencing  134 / 375 ./test_data/test_portrait_images/your_portrait_im/000000251.png\n",
            "--------------------------\n",
            "inferencing  135 / 375 ./test_data/test_portrait_images/your_portrait_im/000000360.png\n",
            "--------------------------\n",
            "inferencing  136 / 375 ./test_data/test_portrait_images/your_portrait_im/000000351.png\n",
            "--------------------------\n",
            "inferencing  137 / 375 ./test_data/test_portrait_images/your_portrait_im/000000342.png\n",
            "--------------------------\n",
            "inferencing  138 / 375 ./test_data/test_portrait_images/your_portrait_im/000000349.png\n",
            "--------------------------\n",
            "inferencing  139 / 375 ./test_data/test_portrait_images/your_portrait_im/000000058.png\n",
            "--------------------------\n",
            "inferencing  140 / 375 ./test_data/test_portrait_images/your_portrait_im/000000361.png\n",
            "--------------------------\n",
            "inferencing  141 / 375 ./test_data/test_portrait_images/your_portrait_im/000000130.png\n",
            "--------------------------\n",
            "inferencing  142 / 375 ./test_data/test_portrait_images/your_portrait_im/000000134.png\n",
            "--------------------------\n",
            "inferencing  143 / 375 ./test_data/test_portrait_images/your_portrait_im/000000350.png\n",
            "--------------------------\n",
            "inferencing  144 / 375 ./test_data/test_portrait_images/your_portrait_im/000000298.png\n",
            "--------------------------\n",
            "inferencing  145 / 375 ./test_data/test_portrait_images/your_portrait_im/000000212.png\n",
            "--------------------------\n",
            "inferencing  146 / 375 ./test_data/test_portrait_images/your_portrait_im/000000032.png\n",
            "--------------------------\n",
            "inferencing  147 / 375 ./test_data/test_portrait_images/your_portrait_im/000000226.png\n",
            "--------------------------\n",
            "inferencing  148 / 375 ./test_data/test_portrait_images/your_portrait_im/000000217.png\n",
            "--------------------------\n",
            "inferencing  149 / 375 ./test_data/test_portrait_images/your_portrait_im/000000201.png\n",
            "--------------------------\n",
            "inferencing  150 / 375 ./test_data/test_portrait_images/your_portrait_im/000000333.png\n",
            "--------------------------\n",
            "inferencing  151 / 375 ./test_data/test_portrait_images/your_portrait_im/000000029.png\n",
            "--------------------------\n",
            "inferencing  152 / 375 ./test_data/test_portrait_images/your_portrait_im/000000012.png\n",
            "--------------------------\n",
            "inferencing  153 / 375 ./test_data/test_portrait_images/your_portrait_im/000000248.png\n",
            "--------------------------\n",
            "inferencing  154 / 375 ./test_data/test_portrait_images/your_portrait_im/000000178.png\n",
            "--------------------------\n",
            "inferencing  155 / 375 ./test_data/test_portrait_images/your_portrait_im/000000205.png\n",
            "--------------------------\n",
            "inferencing  156 / 375 ./test_data/test_portrait_images/your_portrait_im/000000078.png\n",
            "--------------------------\n",
            "inferencing  157 / 375 ./test_data/test_portrait_images/your_portrait_im/000000355.png\n",
            "--------------------------\n",
            "inferencing  158 / 375 ./test_data/test_portrait_images/your_portrait_im/000000300.png\n",
            "--------------------------\n",
            "inferencing  159 / 375 ./test_data/test_portrait_images/your_portrait_im/000000301.png\n",
            "--------------------------\n",
            "inferencing  160 / 375 ./test_data/test_portrait_images/your_portrait_im/000000066.png\n",
            "--------------------------\n",
            "inferencing  161 / 375 ./test_data/test_portrait_images/your_portrait_im/000000150.png\n",
            "--------------------------\n",
            "inferencing  162 / 375 ./test_data/test_portrait_images/your_portrait_im/000000305.png\n",
            "--------------------------\n",
            "inferencing  163 / 375 ./test_data/test_portrait_images/your_portrait_im/000000320.png\n",
            "--------------------------\n",
            "inferencing  164 / 375 ./test_data/test_portrait_images/your_portrait_im/000000101.png\n",
            "--------------------------\n",
            "inferencing  165 / 375 ./test_data/test_portrait_images/your_portrait_im/000000200.png\n",
            "--------------------------\n",
            "inferencing  166 / 375 ./test_data/test_portrait_images/your_portrait_im/000000063.png\n",
            "--------------------------\n",
            "inferencing  167 / 375 ./test_data/test_portrait_images/your_portrait_im/000000104.png\n",
            "--------------------------\n",
            "inferencing  168 / 375 ./test_data/test_portrait_images/your_portrait_im/000000110.png\n",
            "--------------------------\n",
            "inferencing  169 / 375 ./test_data/test_portrait_images/your_portrait_im/000000064.png\n",
            "--------------------------\n",
            "inferencing  170 / 375 ./test_data/test_portrait_images/your_portrait_im/000000002.png\n",
            "--------------------------\n",
            "inferencing  171 / 375 ./test_data/test_portrait_images/your_portrait_im/000000284.png\n",
            "--------------------------\n",
            "inferencing  172 / 375 ./test_data/test_portrait_images/your_portrait_im/000000143.png\n",
            "--------------------------\n",
            "inferencing  173 / 375 ./test_data/test_portrait_images/your_portrait_im/000000325.png\n",
            "--------------------------\n",
            "inferencing  174 / 375 ./test_data/test_portrait_images/your_portrait_im/000000102.png\n",
            "--------------------------\n",
            "inferencing  175 / 375 ./test_data/test_portrait_images/your_portrait_im/000000238.png\n",
            "--------------------------\n",
            "inferencing  176 / 375 ./test_data/test_portrait_images/your_portrait_im/000000057.png\n",
            "--------------------------\n",
            "inferencing  177 / 375 ./test_data/test_portrait_images/your_portrait_im/000000133.png\n",
            "--------------------------\n",
            "inferencing  178 / 375 ./test_data/test_portrait_images/your_portrait_im/000000198.png\n",
            "--------------------------\n",
            "inferencing  179 / 375 ./test_data/test_portrait_images/your_portrait_im/000000331.png\n",
            "--------------------------\n",
            "inferencing  180 / 375 ./test_data/test_portrait_images/your_portrait_im/000000310.png\n",
            "--------------------------\n",
            "inferencing  181 / 375 ./test_data/test_portrait_images/your_portrait_im/000000302.png\n",
            "--------------------------\n",
            "inferencing  182 / 375 ./test_data/test_portrait_images/your_portrait_im/000000275.png\n",
            "--------------------------\n",
            "inferencing  183 / 375 ./test_data/test_portrait_images/your_portrait_im/000000267.png\n",
            "--------------------------\n",
            "inferencing  184 / 375 ./test_data/test_portrait_images/your_portrait_im/000000239.png\n",
            "--------------------------\n",
            "inferencing  185 / 375 ./test_data/test_portrait_images/your_portrait_im/000000336.png\n",
            "--------------------------\n",
            "inferencing  186 / 375 ./test_data/test_portrait_images/your_portrait_im/000000023.png\n",
            "--------------------------\n",
            "inferencing  187 / 375 ./test_data/test_portrait_images/your_portrait_im/000000208.png\n",
            "--------------------------\n",
            "inferencing  188 / 375 ./test_data/test_portrait_images/your_portrait_im/000000254.png\n",
            "--------------------------\n",
            "inferencing  189 / 375 ./test_data/test_portrait_images/your_portrait_im/000000111.png\n",
            "--------------------------\n",
            "inferencing  190 / 375 ./test_data/test_portrait_images/your_portrait_im/000000024.png\n",
            "--------------------------\n",
            "inferencing  191 / 375 ./test_data/test_portrait_images/your_portrait_im/000000340.png\n",
            "--------------------------\n",
            "inferencing  192 / 375 ./test_data/test_portrait_images/your_portrait_im/000000043.png\n",
            "--------------------------\n",
            "inferencing  193 / 375 ./test_data/test_portrait_images/your_portrait_im/000000115.png\n",
            "--------------------------\n",
            "inferencing  194 / 375 ./test_data/test_portrait_images/your_portrait_im/000000108.png\n",
            "--------------------------\n",
            "inferencing  195 / 375 ./test_data/test_portrait_images/your_portrait_im/000000341.png\n",
            "--------------------------\n",
            "inferencing  196 / 375 ./test_data/test_portrait_images/your_portrait_im/000000233.png\n",
            "--------------------------\n",
            "inferencing  197 / 375 ./test_data/test_portrait_images/your_portrait_im/000000292.png\n",
            "--------------------------\n",
            "inferencing  198 / 375 ./test_data/test_portrait_images/your_portrait_im/000000356.png\n",
            "--------------------------\n",
            "inferencing  199 / 375 ./test_data/test_portrait_images/your_portrait_im/000000227.png\n",
            "--------------------------\n",
            "inferencing  200 / 375 ./test_data/test_portrait_images/your_portrait_im/000000147.png\n",
            "--------------------------\n",
            "inferencing  201 / 375 ./test_data/test_portrait_images/your_portrait_im/000000161.png\n",
            "--------------------------\n",
            "inferencing  202 / 375 ./test_data/test_portrait_images/your_portrait_im/000000335.png\n",
            "--------------------------\n",
            "inferencing  203 / 375 ./test_data/test_portrait_images/your_portrait_im/000000145.png\n",
            "--------------------------\n",
            "inferencing  204 / 375 ./test_data/test_portrait_images/your_portrait_im/000000373.png\n",
            "--------------------------\n",
            "inferencing  205 / 375 ./test_data/test_portrait_images/your_portrait_im/000000315.png\n",
            "--------------------------\n",
            "inferencing  206 / 375 ./test_data/test_portrait_images/your_portrait_im/000000347.png\n",
            "--------------------------\n",
            "inferencing  207 / 375 ./test_data/test_portrait_images/your_portrait_im/000000095.png\n",
            "--------------------------\n",
            "inferencing  208 / 375 ./test_data/test_portrait_images/your_portrait_im/000000113.png\n",
            "--------------------------\n",
            "inferencing  209 / 375 ./test_data/test_portrait_images/your_portrait_im/000000307.png\n",
            "--------------------------\n",
            "inferencing  210 / 375 ./test_data/test_portrait_images/your_portrait_im/000000151.png\n",
            "--------------------------\n",
            "inferencing  211 / 375 ./test_data/test_portrait_images/your_portrait_im/000000148.png\n",
            "--------------------------\n",
            "inferencing  212 / 375 ./test_data/test_portrait_images/your_portrait_im/000000228.png\n",
            "--------------------------\n",
            "inferencing  213 / 375 ./test_data/test_portrait_images/your_portrait_im/000000139.png\n",
            "--------------------------\n",
            "inferencing  214 / 375 ./test_data/test_portrait_images/your_portrait_im/000000235.png\n",
            "--------------------------\n",
            "inferencing  215 / 375 ./test_data/test_portrait_images/your_portrait_im/000000138.png\n",
            "--------------------------\n",
            "inferencing  216 / 375 ./test_data/test_portrait_images/your_portrait_im/000000022.png\n",
            "--------------------------\n",
            "inferencing  217 / 375 ./test_data/test_portrait_images/your_portrait_im/000000077.png\n",
            "--------------------------\n",
            "inferencing  218 / 375 ./test_data/test_portrait_images/your_portrait_im/000000169.png\n",
            "--------------------------\n",
            "inferencing  219 / 375 ./test_data/test_portrait_images/your_portrait_im/000000322.png\n",
            "--------------------------\n",
            "inferencing  220 / 375 ./test_data/test_portrait_images/your_portrait_im/000000051.png\n",
            "--------------------------\n",
            "inferencing  221 / 375 ./test_data/test_portrait_images/your_portrait_im/000000344.png\n",
            "--------------------------\n",
            "inferencing  222 / 375 ./test_data/test_portrait_images/your_portrait_im/000000112.png\n",
            "--------------------------\n",
            "inferencing  223 / 375 ./test_data/test_portrait_images/your_portrait_im/000000136.png\n",
            "--------------------------\n",
            "inferencing  224 / 375 ./test_data/test_portrait_images/your_portrait_im/000000128.png\n",
            "--------------------------\n",
            "inferencing  225 / 375 ./test_data/test_portrait_images/your_portrait_im/000000372.png\n",
            "--------------------------\n",
            "inferencing  226 / 375 ./test_data/test_portrait_images/your_portrait_im/000000183.png\n",
            "--------------------------\n",
            "inferencing  227 / 375 ./test_data/test_portrait_images/your_portrait_im/000000120.png\n",
            "--------------------------\n",
            "inferencing  228 / 375 ./test_data/test_portrait_images/your_portrait_im/000000090.png\n",
            "--------------------------\n",
            "inferencing  229 / 375 ./test_data/test_portrait_images/your_portrait_im/000000173.png\n",
            "--------------------------\n",
            "inferencing  230 / 375 ./test_data/test_portrait_images/your_portrait_im/000000240.png\n",
            "--------------------------\n",
            "inferencing  231 / 375 ./test_data/test_portrait_images/your_portrait_im/000000269.png\n",
            "--------------------------\n",
            "inferencing  232 / 375 ./test_data/test_portrait_images/your_portrait_im/000000213.png\n",
            "--------------------------\n",
            "inferencing  233 / 375 ./test_data/test_portrait_images/your_portrait_im/000000368.png\n",
            "--------------------------\n",
            "inferencing  234 / 375 ./test_data/test_portrait_images/your_portrait_im/000000114.png\n",
            "--------------------------\n",
            "inferencing  235 / 375 ./test_data/test_portrait_images/your_portrait_im/000000296.png\n",
            "--------------------------\n",
            "inferencing  236 / 375 ./test_data/test_portrait_images/your_portrait_im/000000209.png\n",
            "--------------------------\n",
            "inferencing  237 / 375 ./test_data/test_portrait_images/your_portrait_im/000000049.png\n",
            "--------------------------\n",
            "inferencing  238 / 375 ./test_data/test_portrait_images/your_portrait_im/000000367.png\n",
            "--------------------------\n",
            "inferencing  239 / 375 ./test_data/test_portrait_images/your_portrait_im/000000125.png\n",
            "--------------------------\n",
            "inferencing  240 / 375 ./test_data/test_portrait_images/your_portrait_im/000000001.png\n",
            "--------------------------\n",
            "inferencing  241 / 375 ./test_data/test_portrait_images/your_portrait_im/000000283.png\n",
            "--------------------------\n",
            "inferencing  242 / 375 ./test_data/test_portrait_images/your_portrait_im/000000179.png\n",
            "--------------------------\n",
            "inferencing  243 / 375 ./test_data/test_portrait_images/your_portrait_im/000000288.png\n",
            "--------------------------\n",
            "inferencing  244 / 375 ./test_data/test_portrait_images/your_portrait_im/000000272.png\n",
            "--------------------------\n",
            "inferencing  245 / 375 ./test_data/test_portrait_images/your_portrait_im/000000323.png\n",
            "--------------------------\n",
            "inferencing  246 / 375 ./test_data/test_portrait_images/your_portrait_im/000000082.png\n",
            "--------------------------\n",
            "inferencing  247 / 375 ./test_data/test_portrait_images/your_portrait_im/000000230.png\n",
            "--------------------------\n",
            "inferencing  248 / 375 ./test_data/test_portrait_images/your_portrait_im/000000324.png\n",
            "--------------------------\n",
            "inferencing  249 / 375 ./test_data/test_portrait_images/your_portrait_im/000000365.png\n",
            "--------------------------\n",
            "inferencing  250 / 375 ./test_data/test_portrait_images/your_portrait_im/000000199.png\n",
            "--------------------------\n",
            "inferencing  251 / 375 ./test_data/test_portrait_images/your_portrait_im/000000149.png\n",
            "--------------------------\n",
            "inferencing  252 / 375 ./test_data/test_portrait_images/your_portrait_im/000000343.png\n",
            "--------------------------\n",
            "inferencing  253 / 375 ./test_data/test_portrait_images/your_portrait_im/000000011.png\n",
            "--------------------------\n",
            "inferencing  254 / 375 ./test_data/test_portrait_images/your_portrait_im/000000109.png\n",
            "--------------------------\n",
            "inferencing  255 / 375 ./test_data/test_portrait_images/your_portrait_im/000000196.png\n",
            "--------------------------\n",
            "inferencing  256 / 375 ./test_data/test_portrait_images/your_portrait_im/000000291.png\n",
            "--------------------------\n",
            "inferencing  257 / 375 ./test_data/test_portrait_images/your_portrait_im/000000223.png\n",
            "--------------------------\n",
            "inferencing  258 / 375 ./test_data/test_portrait_images/your_portrait_im/000000146.png\n",
            "--------------------------\n",
            "inferencing  259 / 375 ./test_data/test_portrait_images/your_portrait_im/000000311.png\n",
            "--------------------------\n",
            "inferencing  260 / 375 ./test_data/test_portrait_images/your_portrait_im/000000059.png\n",
            "--------------------------\n",
            "inferencing  261 / 375 ./test_data/test_portrait_images/your_portrait_im/000000318.png\n",
            "--------------------------\n",
            "inferencing  262 / 375 ./test_data/test_portrait_images/your_portrait_im/000000338.png\n",
            "--------------------------\n",
            "inferencing  263 / 375 ./test_data/test_portrait_images/your_portrait_im/000000048.png\n",
            "--------------------------\n",
            "inferencing  264 / 375 ./test_data/test_portrait_images/your_portrait_im/000000188.png\n",
            "--------------------------\n",
            "inferencing  265 / 375 ./test_data/test_portrait_images/your_portrait_im/000000073.png\n",
            "--------------------------\n",
            "inferencing  266 / 375 ./test_data/test_portrait_images/your_portrait_im/000000187.png\n",
            "--------------------------\n",
            "inferencing  267 / 375 ./test_data/test_portrait_images/your_portrait_im/000000061.png\n",
            "--------------------------\n",
            "inferencing  268 / 375 ./test_data/test_portrait_images/your_portrait_im/000000362.png\n",
            "--------------------------\n",
            "inferencing  269 / 375 ./test_data/test_portrait_images/your_portrait_im/000000192.png\n",
            "--------------------------\n",
            "inferencing  270 / 375 ./test_data/test_portrait_images/your_portrait_im/000000234.png\n",
            "--------------------------\n",
            "inferencing  271 / 375 ./test_data/test_portrait_images/your_portrait_im/000000099.png\n",
            "--------------------------\n",
            "inferencing  272 / 375 ./test_data/test_portrait_images/your_portrait_im/000000206.png\n",
            "--------------------------\n",
            "inferencing  273 / 375 ./test_data/test_portrait_images/your_portrait_im/000000156.png\n",
            "--------------------------\n",
            "inferencing  274 / 375 ./test_data/test_portrait_images/your_portrait_im/000000222.png\n",
            "--------------------------\n",
            "inferencing  275 / 375 ./test_data/test_portrait_images/your_portrait_im/000000308.png\n",
            "--------------------------\n",
            "inferencing  276 / 375 ./test_data/test_portrait_images/your_portrait_im/000000181.png\n",
            "--------------------------\n",
            "inferencing  277 / 375 ./test_data/test_portrait_images/your_portrait_im/000000096.png\n",
            "--------------------------\n",
            "inferencing  278 / 375 ./test_data/test_portrait_images/your_portrait_im/000000354.png\n",
            "--------------------------\n",
            "inferencing  279 / 375 ./test_data/test_portrait_images/your_portrait_im/000000319.png\n",
            "--------------------------\n",
            "inferencing  280 / 375 ./test_data/test_portrait_images/your_portrait_im/000000295.png\n",
            "--------------------------\n",
            "inferencing  281 / 375 ./test_data/test_portrait_images/your_portrait_im/000000170.png\n",
            "--------------------------\n",
            "inferencing  282 / 375 ./test_data/test_portrait_images/your_portrait_im/000000328.png\n",
            "--------------------------\n",
            "inferencing  283 / 375 ./test_data/test_portrait_images/your_portrait_im/000000054.png\n",
            "--------------------------\n",
            "inferencing  284 / 375 ./test_data/test_portrait_images/your_portrait_im/000000135.png\n",
            "--------------------------\n",
            "inferencing  285 / 375 ./test_data/test_portrait_images/your_portrait_im/000000262.png\n",
            "--------------------------\n",
            "inferencing  286 / 375 ./test_data/test_portrait_images/your_portrait_im/000000155.png\n",
            "--------------------------\n",
            "inferencing  287 / 375 ./test_data/test_portrait_images/your_portrait_im/000000035.png\n",
            "--------------------------\n",
            "inferencing  288 / 375 ./test_data/test_portrait_images/your_portrait_im/000000079.png\n",
            "--------------------------\n",
            "inferencing  289 / 375 ./test_data/test_portrait_images/your_portrait_im/000000162.png\n",
            "--------------------------\n",
            "inferencing  290 / 375 ./test_data/test_portrait_images/your_portrait_im/000000039.png\n",
            "--------------------------\n",
            "inferencing  291 / 375 ./test_data/test_portrait_images/your_portrait_im/000000137.png\n",
            "--------------------------\n",
            "inferencing  292 / 375 ./test_data/test_portrait_images/your_portrait_im/000000119.png\n",
            "--------------------------\n",
            "inferencing  293 / 375 ./test_data/test_portrait_images/your_portrait_im/000000123.png\n",
            "--------------------------\n",
            "inferencing  294 / 375 ./test_data/test_portrait_images/your_portrait_im/000000055.png\n",
            "--------------------------\n",
            "inferencing  295 / 375 ./test_data/test_portrait_images/your_portrait_im/000000042.png\n",
            "--------------------------\n",
            "inferencing  296 / 375 ./test_data/test_portrait_images/your_portrait_im/000000165.png\n",
            "--------------------------\n",
            "inferencing  297 / 375 ./test_data/test_portrait_images/your_portrait_im/000000168.png\n",
            "--------------------------\n",
            "inferencing  298 / 375 ./test_data/test_portrait_images/your_portrait_im/000000241.png\n",
            "--------------------------\n",
            "inferencing  299 / 375 ./test_data/test_portrait_images/your_portrait_im/000000158.png\n",
            "--------------------------\n",
            "inferencing  300 / 375 ./test_data/test_portrait_images/your_portrait_im/000000028.png\n",
            "--------------------------\n",
            "inferencing  301 / 375 ./test_data/test_portrait_images/your_portrait_im/000000071.png\n",
            "--------------------------\n",
            "inferencing  302 / 375 ./test_data/test_portrait_images/your_portrait_im/000000202.png\n",
            "--------------------------\n",
            "inferencing  303 / 375 ./test_data/test_portrait_images/your_portrait_im/000000016.png\n",
            "--------------------------\n",
            "inferencing  304 / 375 ./test_data/test_portrait_images/your_portrait_im/000000020.png\n",
            "--------------------------\n",
            "inferencing  305 / 375 ./test_data/test_portrait_images/your_portrait_im/000000065.png\n",
            "--------------------------\n",
            "inferencing  306 / 375 ./test_data/test_portrait_images/your_portrait_im/000000166.png\n",
            "--------------------------\n",
            "inferencing  307 / 375 ./test_data/test_portrait_images/your_portrait_im/000000087.png\n",
            "--------------------------\n",
            "inferencing  308 / 375 ./test_data/test_portrait_images/your_portrait_im/000000021.png\n",
            "--------------------------\n",
            "inferencing  309 / 375 ./test_data/test_portrait_images/your_portrait_im/000000211.png\n",
            "--------------------------\n",
            "inferencing  310 / 375 ./test_data/test_portrait_images/your_portrait_im/000000245.png\n",
            "--------------------------\n",
            "inferencing  311 / 375 ./test_data/test_portrait_images/your_portrait_im/000000304.png\n",
            "--------------------------\n",
            "inferencing  312 / 375 ./test_data/test_portrait_images/your_portrait_im/000000219.png\n",
            "--------------------------\n",
            "inferencing  313 / 375 ./test_data/test_portrait_images/your_portrait_im/000000264.png\n",
            "--------------------------\n",
            "inferencing  314 / 375 ./test_data/test_portrait_images/your_portrait_im/000000000.png\n",
            "--------------------------\n",
            "inferencing  315 / 375 ./test_data/test_portrait_images/your_portrait_im/000000327.png\n",
            "--------------------------\n",
            "inferencing  316 / 375 ./test_data/test_portrait_images/your_portrait_im/000000329.png\n",
            "--------------------------\n",
            "inferencing  317 / 375 ./test_data/test_portrait_images/your_portrait_im/000000033.png\n",
            "--------------------------\n",
            "inferencing  318 / 375 ./test_data/test_portrait_images/your_portrait_im/000000231.png\n",
            "--------------------------\n",
            "inferencing  319 / 375 ./test_data/test_portrait_images/your_portrait_im/000000034.png\n",
            "--------------------------\n",
            "inferencing  320 / 375 ./test_data/test_portrait_images/your_portrait_im/000000081.png\n",
            "--------------------------\n",
            "inferencing  321 / 375 ./test_data/test_portrait_images/your_portrait_im/000000121.png\n",
            "--------------------------\n",
            "inferencing  322 / 375 ./test_data/test_portrait_images/your_portrait_im/000000274.png\n",
            "--------------------------\n",
            "inferencing  323 / 375 ./test_data/test_portrait_images/your_portrait_im/000000260.png\n",
            "--------------------------\n",
            "inferencing  324 / 375 ./test_data/test_portrait_images/your_portrait_im/000000186.png\n",
            "--------------------------\n",
            "inferencing  325 / 375 ./test_data/test_portrait_images/your_portrait_im/000000152.png\n",
            "--------------------------\n",
            "inferencing  326 / 375 ./test_data/test_portrait_images/your_portrait_im/000000263.png\n",
            "--------------------------\n",
            "inferencing  327 / 375 ./test_data/test_portrait_images/your_portrait_im/000000334.png\n",
            "--------------------------\n",
            "inferencing  328 / 375 ./test_data/test_portrait_images/your_portrait_im/000000193.png\n",
            "--------------------------\n",
            "inferencing  329 / 375 ./test_data/test_portrait_images/your_portrait_im/000000216.png\n",
            "--------------------------\n",
            "inferencing  330 / 375 ./test_data/test_portrait_images/your_portrait_im/000000103.png\n",
            "--------------------------\n",
            "inferencing  331 / 375 ./test_data/test_portrait_images/your_portrait_im/000000185.png\n",
            "--------------------------\n",
            "inferencing  332 / 375 ./test_data/test_portrait_images/your_portrait_im/000000237.png\n",
            "--------------------------\n",
            "inferencing  333 / 375 ./test_data/test_portrait_images/your_portrait_im/000000253.png\n",
            "--------------------------\n",
            "inferencing  334 / 375 ./test_data/test_portrait_images/your_portrait_im/000000046.png\n",
            "--------------------------\n",
            "inferencing  335 / 375 ./test_data/test_portrait_images/your_portrait_im/000000067.png\n",
            "--------------------------\n",
            "inferencing  336 / 375 ./test_data/test_portrait_images/your_portrait_im/000000129.png\n",
            "--------------------------\n",
            "inferencing  337 / 375 ./test_data/test_portrait_images/your_portrait_im/000000159.png\n",
            "--------------------------\n",
            "inferencing  338 / 375 ./test_data/test_portrait_images/your_portrait_im/000000306.png\n",
            "--------------------------\n",
            "inferencing  339 / 375 ./test_data/test_portrait_images/your_portrait_im/000000160.png\n",
            "--------------------------\n",
            "inferencing  340 / 375 ./test_data/test_portrait_images/your_portrait_im/000000346.png\n",
            "--------------------------\n",
            "inferencing  341 / 375 ./test_data/test_portrait_images/your_portrait_im/000000236.png\n",
            "--------------------------\n",
            "inferencing  342 / 375 ./test_data/test_portrait_images/your_portrait_im/000000084.png\n",
            "--------------------------\n",
            "inferencing  343 / 375 ./test_data/test_portrait_images/your_portrait_im/000000276.png\n",
            "--------------------------\n",
            "inferencing  344 / 375 ./test_data/test_portrait_images/your_portrait_im/000000176.png\n",
            "--------------------------\n",
            "inferencing  345 / 375 ./test_data/test_portrait_images/your_portrait_im/000000278.png\n",
            "--------------------------\n",
            "inferencing  346 / 375 ./test_data/test_portrait_images/your_portrait_im/000000250.png\n",
            "--------------------------\n",
            "inferencing  347 / 375 ./test_data/test_portrait_images/your_portrait_im/000000257.png\n",
            "--------------------------\n",
            "inferencing  348 / 375 ./test_data/test_portrait_images/your_portrait_im/000000369.png\n",
            "--------------------------\n",
            "inferencing  349 / 375 ./test_data/test_portrait_images/your_portrait_im/000000141.png\n",
            "--------------------------\n",
            "inferencing  350 / 375 ./test_data/test_portrait_images/your_portrait_im/000000132.png\n",
            "--------------------------\n",
            "inferencing  351 / 375 ./test_data/test_portrait_images/your_portrait_im/000000118.png\n",
            "--------------------------\n",
            "inferencing  352 / 375 ./test_data/test_portrait_images/your_portrait_im/000000080.png\n",
            "--------------------------\n",
            "inferencing  353 / 375 ./test_data/test_portrait_images/your_portrait_im/000000191.png\n",
            "--------------------------\n",
            "inferencing  354 / 375 ./test_data/test_portrait_images/your_portrait_im/000000190.png\n",
            "--------------------------\n",
            "inferencing  355 / 375 ./test_data/test_portrait_images/your_portrait_im/000000083.png\n",
            "--------------------------\n",
            "inferencing  356 / 375 ./test_data/test_portrait_images/your_portrait_im/000000180.png\n",
            "--------------------------\n",
            "inferencing  357 / 375 ./test_data/test_portrait_images/your_portrait_im/000000060.png\n",
            "--------------------------\n",
            "inferencing  358 / 375 ./test_data/test_portrait_images/your_portrait_im/000000098.png\n",
            "--------------------------\n",
            "inferencing  359 / 375 ./test_data/test_portrait_images/your_portrait_im/000000297.png\n",
            "--------------------------\n",
            "inferencing  360 / 375 ./test_data/test_portrait_images/your_portrait_im/000000094.png\n",
            "--------------------------\n",
            "inferencing  361 / 375 ./test_data/test_portrait_images/your_portrait_im/000000268.png\n",
            "--------------------------\n",
            "inferencing  362 / 375 ./test_data/test_portrait_images/your_portrait_im/000000194.png\n",
            "--------------------------\n",
            "inferencing  363 / 375 ./test_data/test_portrait_images/your_portrait_im/000000053.png\n",
            "--------------------------\n",
            "inferencing  364 / 375 ./test_data/test_portrait_images/your_portrait_im/000000280.png\n",
            "--------------------------\n",
            "inferencing  365 / 375 ./test_data/test_portrait_images/your_portrait_im/000000259.png\n",
            "--------------------------\n",
            "inferencing  366 / 375 ./test_data/test_portrait_images/your_portrait_im/000000321.png\n",
            "--------------------------\n",
            "inferencing  367 / 375 ./test_data/test_portrait_images/your_portrait_im/000000093.png\n",
            "--------------------------\n",
            "inferencing  368 / 375 ./test_data/test_portrait_images/your_portrait_im/000000174.png\n",
            "--------------------------\n",
            "inferencing  369 / 375 ./test_data/test_portrait_images/your_portrait_im/000000303.png\n",
            "--------------------------\n",
            "inferencing  370 / 375 ./test_data/test_portrait_images/your_portrait_im/000000052.png\n",
            "--------------------------\n",
            "inferencing  371 / 375 ./test_data/test_portrait_images/your_portrait_im/000000243.png\n",
            "--------------------------\n",
            "inferencing  372 / 375 ./test_data/test_portrait_images/your_portrait_im/000000332.png\n",
            "--------------------------\n",
            "inferencing  373 / 375 ./test_data/test_portrait_images/your_portrait_im/000000026.png\n",
            "--------------------------\n",
            "inferencing  374 / 375 ./test_data/test_portrait_images/your_portrait_im/000000050.png\n",
            "Generate video\n",
            "ffmpeg version 3.4.8-0ubuntu0.2 Copyright (c) 2000-2020 the FFmpeg developers\n",
            "  built with gcc 7 (Ubuntu 7.5.0-3ubuntu1~18.04)\n",
            "  configuration: --prefix=/usr --extra-version=0ubuntu0.2 --toolchain=hardened --libdir=/usr/lib/x86_64-linux-gnu --incdir=/usr/include/x86_64-linux-gnu --enable-gpl --disable-stripping --enable-avresample --enable-avisynth --enable-gnutls --enable-ladspa --enable-libass --enable-libbluray --enable-libbs2b --enable-libcaca --enable-libcdio --enable-libflite --enable-libfontconfig --enable-libfreetype --enable-libfribidi --enable-libgme --enable-libgsm --enable-libmp3lame --enable-libmysofa --enable-libopenjpeg --enable-libopenmpt --enable-libopus --enable-libpulse --enable-librubberband --enable-librsvg --enable-libshine --enable-libsnappy --enable-libsoxr --enable-libspeex --enable-libssh --enable-libtheora --enable-libtwolame --enable-libvorbis --enable-libvpx --enable-libwavpack --enable-libwebp --enable-libx265 --enable-libxml2 --enable-libxvid --enable-libzmq --enable-libzvbi --enable-omx --enable-openal --enable-opengl --enable-sdl2 --enable-libdc1394 --enable-libdrm --enable-libiec61883 --enable-chromaprint --enable-frei0r --enable-libopencv --enable-libx264 --enable-shared\n",
            "  libavutil      55. 78.100 / 55. 78.100\n",
            "  libavcodec     57.107.100 / 57.107.100\n",
            "  libavformat    57. 83.100 / 57. 83.100\n",
            "  libavdevice    57. 10.100 / 57. 10.100\n",
            "  libavfilter     6.107.100 /  6.107.100\n",
            "  libavresample   3.  7.  0 /  3.  7.  0\n",
            "  libswscale      4.  8.100 /  4.  8.100\n",
            "  libswresample   2.  9.100 /  2.  9.100\n",
            "  libpostproc    54.  7.100 / 54.  7.100\n",
            "Input #0, image2, from '/content/U-2-Net/test_data/test_portrait_images/your_portrait_results/%09d.png':\n",
            "  Duration: 00:00:15.00, start: 0.000000, bitrate: N/A\n",
            "    Stream #0:0: Video: png, gray(pc), 1080x720, 25 fps, 25 tbr, 25 tbn, 25 tbc\n",
            "Stream mapping:\n",
            "  Stream #0:0 -> #0:0 (png (native) -> prores (native))\n",
            "Press [q] to stop, [?] for help\n",
            "\u001b[1;36m[prores @ 0x5638c8afde00] \u001b[0mencoding with ProRes standard (apcn) profile\n",
            "\u001b[1;36m[prores @ 0x5638c8afe300] \u001b[0mencoding with ProRes standard (apcn) profile\n",
            "\u001b[1;36m[prores @ 0x5638c83dbe00] \u001b[0mencoding with ProRes standard (apcn) profile\n",
            "Output #0, mp4, to '/content/U-2-Net/result.mp4':\n",
            "  Metadata:\n",
            "    encoder         : Lavf57.83.100\n",
            "    Stream #0:0: Video: prores (apcn / 0x6E637061), yuv422p10le, 1080x720, q=2-31, 200 kb/s, 25 fps, 12800 tbn, 25 tbc\n",
            "    Metadata:\n",
            "      encoder         : Lavc57.107.100 prores\n",
            "frame=  375 fps= 30 q=-0.0 Lsize=   91571kB time=00:00:14.96 bitrate=50143.4kbits/s speed=1.18x    \n",
            "video:91568kB audio:0kB subtitle:0kB other streams:0kB global headers:0kB muxing overhead: 0.002921%\n",
            "ready\n"
          ],
          "name": "stdout"
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "cXUCqtAyjWtZ",
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 53
        },
        "outputId": "e1426ac5-69fd-4990-9f51-641c47bc60f3"
      },
      "source": [
        "#@title <b><font color=\"red\" size=\"+3\">←</font><font color=\"black\" size=\"+3\"> Скачиваем итоговый файл</font></b>\n",
        "!ls -l $path_to_res\n",
        "files.download(path_to_res)"
      ],
      "execution_count": 15,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "-rw-r--r-- 1 root root 93768705 Aug 20 08:59 /content/U-2-Net/result.mp4\n"
          ],
          "name": "stdout"
        },
        {
          "output_type": "display_data",
          "data": {
            "application/javascript": [
              "\n",
              "    async function download(id, filename, size) {\n",
              "      if (!google.colab.kernel.accessAllowed) {\n",
              "        return;\n",
              "      }\n",
              "      const div = document.createElement('div');\n",
              "      const label = document.createElement('label');\n",
              "      label.textContent = `Downloading \"${filename}\": `;\n",
              "      div.appendChild(label);\n",
              "      const progress = document.createElement('progress');\n",
              "      progress.max = size;\n",
              "      div.appendChild(progress);\n",
              "      document.body.appendChild(div);\n",
              "\n",
              "      const buffers = [];\n",
              "      let downloaded = 0;\n",
              "\n",
              "      const channel = await google.colab.kernel.comms.open(id);\n",
              "      // Send a message to notify the kernel that we're ready.\n",
              "      channel.send({})\n",
              "\n",
              "      for await (const message of channel.messages) {\n",
              "        // Send a message to notify the kernel that we're ready.\n",
              "        channel.send({})\n",
              "        if (message.buffers) {\n",
              "          for (const buffer of message.buffers) {\n",
              "            buffers.push(buffer);\n",
              "            downloaded += buffer.byteLength;\n",
              "            progress.value = downloaded;\n",
              "          }\n",
              "        }\n",
              "      }\n",
              "      const blob = new Blob(buffers, {type: 'application/binary'});\n",
              "      const a = document.createElement('a');\n",
              "      a.href = window.URL.createObjectURL(blob);\n",
              "      a.download = filename;\n",
              "      div.appendChild(a);\n",
              "      a.click();\n",
              "      div.remove();\n",
              "    }\n",
              "  "
            ],
            "text/plain": [
              "<IPython.core.display.Javascript object>"
            ]
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "application/javascript": [
              "download(\"download_3e1a70f5-b3f4-4304-a0a9-3bae45f92bc0\", \"result.mp4\", 93768705)"
            ],
            "text/plain": [
              "<IPython.core.display.Javascript object>"
            ]
          },
          "metadata": {}
        }
      ]
    }
  ]
}