Skip to content

Instantly share code, notes, and snippets.

@duskvirkus

duskvirkus/sg3-bug.ipynb Secret

Created October 12, 2021 04:38
Show Gist options
  • Save duskvirkus/46257dfc709854577128c0a04ffda987 to your computer and use it in GitHub Desktop.
Save duskvirkus/46257dfc709854577128c0a04ffda987 to your computer and use it in GitHub Desktop.
Download ZIP
sg3-bug?.ipynb
Display the source blob
Display the rendered blob
Raw
sg3-bug.ipynb
{
"nbformat": 4,
"nbformat_minor": 0,
"metadata": {
"colab": {
"name": "sg3-bug?.ipynb",
"provenance": [],
"collapsed_sections": [],
"authorship_tag": "ABX9TyMupv6Z9BW6cLC0TKS7vAbq",
"include_colab_link": true
},
"kernelspec": {
"name": "python3",
"display_name": "Python 3"
},
"language_info": {
"name": "python"
},
"accelerator": "GPU"
},
"cells": [
{
"cell_type": "markdown",
"metadata": {
"id": "view-in-github",
"colab_type": "text"
},
"source": [
"<a href=\"https://colab.research.google.com/gist/duskvirkus/46257dfc709854577128c0a04ffda987/sg3-bug.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "1IJZZUU9iIL3"
},
"source": [
"## Setup"
]
},
{
"cell_type": "code",
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "SuJlRDvkgnWX",
"outputId": "1d093900-f364-4a0e-90df-252abce9a0ad"
},
"source": [
"!nvidia-smi -L"
],
"execution_count": null,
"outputs": [
{
"output_type": "stream",
"name": "stdout",
"text": [
"GPU 0: Tesla P100-PCIE-16GB (UUID: GPU-f15f17d1-2f5e-d3e0-3edf-d60006851f4d)\n"
]
}
]
},
{
"cell_type": "code",
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "ngotbSQpf0Wk",
"outputId": "a2136fe4-ec42-44e5-f520-9073838a243b"
},
"source": [
"from google.colab import drive\n",
"drive.mount('/content/drive')"
],
"execution_count": null,
"outputs": [
{
"output_type": "stream",
"name": "stdout",
"text": [
"Mounted at /content/drive\n"
]
}
]
},
{
"cell_type": "code",
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "fBYpgveVgNuY",
"outputId": "2b25d00f-c54c-4b69-da93-5ea9cb73aded"
},
"source": [
"import os\n",
"drive_path = '/content/drive/MyDrive/'\n",
"repo_container_dir = 'colab-sg3-super-res'\n",
"repo_name = 'stylegan3'\n",
"git_repo = 'https://github.com/duskvirkus/stylegan3.git'\n",
"git_branch = 'main'\n",
"\n",
"working_dir = os.path.join(drive_path, repo_container_dir, repo_name)\n",
"\n",
"if os.path.isdir(working_dir):\n",
" %cd {working_dir}\n",
"else:\n",
" container_path = os.path.join(drive_path, repo_container_dir)\n",
" os.makedirs(container_path)\n",
" %cd {container_path}\n",
" !git clone --branch {git_branch} {git_repo}\n",
" %cd {repo_name}\n",
" !mkdir pretrained"
],
"execution_count": null,
"outputs": [
{
"output_type": "stream",
"name": "stdout",
"text": [
"/content/drive/MyDrive/colab-sg3-super-res\n",
"Cloning into 'stylegan3'...\n",
"remote: Enumerating objects: 93, done.\u001b[K\n",
"remote: Counting objects: 100% (93/93), done.\u001b[K\n",
"remote: Compressing objects: 100% (80/80), done.\u001b[K\n",
"remote: Total 93 (delta 12), reused 90 (delta 11), pack-reused 0\u001b[K\n",
"Unpacking objects: 100% (93/93), done.\n",
"/content/drive/MyDrive/colab-sg3-super-res/stylegan3\n"
]
}
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "T0pbzPc0iO-B"
},
"source": [
"## Download MetFaces Model"
]
},
{
"cell_type": "code",
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "wv1GXrhQijFj",
"outputId": "108539fd-4890-41f0-de3b-233844303240"
},
"source": [
"!wget https://api.ngc.nvidia.com/v2/models/nvidia/research/stylegan3/versions/1/files/stylegan3-t-metfacesu-1024x1024.pkl"
],
"execution_count": null,
"outputs": [
{
"output_type": "stream",
"name": "stdout",
"text": [
"--2021-10-12 04:17:54-- https://api.ngc.nvidia.com/v2/models/nvidia/research/stylegan3/versions/1/files/stylegan3-t-metfacesu-1024x1024.pkl\n",
"Resolving api.ngc.nvidia.com (api.ngc.nvidia.com)... 44.232.207.17, 44.238.121.239\n",
"Connecting to api.ngc.nvidia.com (api.ngc.nvidia.com)|44.232.207.17|:443... connected.\n",
"HTTP request sent, awaiting response... 302 \n",
"Location: https://prod-model-registry-ngc-bucket.s3.us-west-2.amazonaws.com/org/nvidia/team/research/models/stylegan3/versions/1/files/stylegan3-t-metfacesu-1024x1024.pkl?response-content-disposition=attachment%3B%20filename%3D%22stylegan3-t-metfacesu-1024x1024.pkl%22&response-content-type=application%2Foctet-stream&X-Amz-Security-Token=IQoJb3JpZ2luX2VjEIP%2F%2F%2F%2F%2F%2F%2F%2F%2F%2FwEaCXVzLXdlc3QtMiJHMEUCIQDTmdaWBTWuPhSQlFS3vpcDMdySS611o5aJ6tQnEJl5MAIgQtRWHP21mHo0uBXedkgmuA0lZUjYaONJrNaKhS3Eyg4qgwQI%2FP%2F%2F%2F%2F%2F%2F%2F%2F%2F%2FARADGgw3ODkzNjMxMzUwMjciDHOCWR3JEq1rbBSZhSrXA3dMVhIuSMngssjfLGBEuzTvz%2BElgJ5SyBXytTFOs1xOCtOUymicpNi%2Bb0oVv4awJPGyFx8Lf5L%2F7UGv8jMLXmgaFArK7a2bgj5MMpIky85cSx6mrA1rqCe%2FOaBXPmEqtzcS14khocT%2Fp6KbzQoMnOV4hddfQ0sjc9H0bXOg5GTtssZZ62r1bfmV3NpfKp5FomnH7LXHaeZ73tBbz1vScjyCIg91XNOB1T4eHYZ5tEBhcCFtI9p5rBj5%2FhJKhxDNHZXPSsUYHZRrRbDzT2RYPUtLOt6DURDiYdfhkoIPrAFOHsRrUQggGR3OwCzgE7g7h9z04frfrCRiPo6YlaYZMXYa7QyyNgQitaImt2lAbR2jnmK9FQjyT4uyHAZYUKySARnjT7vWsNJmwmYUyNVIFvHkMj6f1oRShkFebWkEcmaXKJyY%2F7xU1wnX2zHgkDWHGbvfVUVPkJWNQ6%2BfRF%2B4EIoAVGEXy2Gqj6A2hDhvdbqBBkDGdCaRguQ84XLA0BoTxlyMAZZMrYtK6eEEUYjvv%2BXz%2BUh87vL1%2B9Vr20EESHVYse3i5Y49I%2BNeoopx38wTBmDuqD2mCv%2BAcYZO8h1d2h7MOar0%2Fhc%2FDKwnVRZu8MYlOr4ps3QE%2FTCn%2BpOLBjqlAZptffkyL3rbvDdFihKlEJ1Y70qrddYA%2BPCFuRL7fjV5Fl27TOXK1GqNML%2F%2FsbUtLhnkHBE5c%2F3tPHhCzdo%2Bh4advqj%2FmyCBiRsLfoAkSB%2Fv90HKZoEXw8hjRtxODT0c7mSXZgnYkX5baHFA7z%2BWsDPWwJMOVzDjKZ6cRBMYON8fvh7L3C82QqoacCGqmpodhnC6YqYocQ4tQH2c6v57YH%2BRYQsvog%3D%3D&X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Date=20211012T041754Z&X-Amz-SignedHeaders=host&X-Amz-Expires=3600&X-Amz-Credential=ASIA3PSNVSIZUCANAJSS%2F20211012%2Fus-west-2%2Fs3%2Faws4_request&X-Amz-Signature=9ac46a6ba1e8b632a4c28f8d3bad333983f448ebcda839217435afda12e80629 [following]\n",
"--2021-10-12 04:17:54-- https://prod-model-registry-ngc-bucket.s3.us-west-2.amazonaws.com/org/nvidia/team/research/models/stylegan3/versions/1/files/stylegan3-t-metfacesu-1024x1024.pkl?response-content-disposition=attachment%3B%20filename%3D%22stylegan3-t-metfacesu-1024x1024.pkl%22&response-content-type=application%2Foctet-stream&X-Amz-Security-Token=IQoJb3JpZ2luX2VjEIP%2F%2F%2F%2F%2F%2F%2F%2F%2F%2FwEaCXVzLXdlc3QtMiJHMEUCIQDTmdaWBTWuPhSQlFS3vpcDMdySS611o5aJ6tQnEJl5MAIgQtRWHP21mHo0uBXedkgmuA0lZUjYaONJrNaKhS3Eyg4qgwQI%2FP%2F%2F%2F%2F%2F%2F%2F%2F%2F%2FARADGgw3ODkzNjMxMzUwMjciDHOCWR3JEq1rbBSZhSrXA3dMVhIuSMngssjfLGBEuzTvz%2BElgJ5SyBXytTFOs1xOCtOUymicpNi%2Bb0oVv4awJPGyFx8Lf5L%2F7UGv8jMLXmgaFArK7a2bgj5MMpIky85cSx6mrA1rqCe%2FOaBXPmEqtzcS14khocT%2Fp6KbzQoMnOV4hddfQ0sjc9H0bXOg5GTtssZZ62r1bfmV3NpfKp5FomnH7LXHaeZ73tBbz1vScjyCIg91XNOB1T4eHYZ5tEBhcCFtI9p5rBj5%2FhJKhxDNHZXPSsUYHZRrRbDzT2RYPUtLOt6DURDiYdfhkoIPrAFOHsRrUQggGR3OwCzgE7g7h9z04frfrCRiPo6YlaYZMXYa7QyyNgQitaImt2lAbR2jnmK9FQjyT4uyHAZYUKySARnjT7vWsNJmwmYUyNVIFvHkMj6f1oRShkFebWkEcmaXKJyY%2F7xU1wnX2zHgkDWHGbvfVUVPkJWNQ6%2BfRF%2B4EIoAVGEXy2Gqj6A2hDhvdbqBBkDGdCaRguQ84XLA0BoTxlyMAZZMrYtK6eEEUYjvv%2BXz%2BUh87vL1%2B9Vr20EESHVYse3i5Y49I%2BNeoopx38wTBmDuqD2mCv%2BAcYZO8h1d2h7MOar0%2Fhc%2FDKwnVRZu8MYlOr4ps3QE%2FTCn%2BpOLBjqlAZptffkyL3rbvDdFihKlEJ1Y70qrddYA%2BPCFuRL7fjV5Fl27TOXK1GqNML%2F%2FsbUtLhnkHBE5c%2F3tPHhCzdo%2Bh4advqj%2FmyCBiRsLfoAkSB%2Fv90HKZoEXw8hjRtxODT0c7mSXZgnYkX5baHFA7z%2BWsDPWwJMOVzDjKZ6cRBMYON8fvh7L3C82QqoacCGqmpodhnC6YqYocQ4tQH2c6v57YH%2BRYQsvog%3D%3D&X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Date=20211012T041754Z&X-Amz-SignedHeaders=host&X-Amz-Expires=3600&X-Amz-Credential=ASIA3PSNVSIZUCANAJSS%2F20211012%2Fus-west-2%2Fs3%2Faws4_request&X-Amz-Signature=9ac46a6ba1e8b632a4c28f8d3bad333983f448ebcda839217435afda12e80629\n",
"Resolving prod-model-registry-ngc-bucket.s3.us-west-2.amazonaws.com (prod-model-registry-ngc-bucket.s3.us-west-2.amazonaws.com)... 52.218.178.73\n",
"Connecting to prod-model-registry-ngc-bucket.s3.us-west-2.amazonaws.com (prod-model-registry-ngc-bucket.s3.us-west-2.amazonaws.com)|52.218.178.73|:443... connected.\n",
"HTTP request sent, awaiting response... 200 OK\n",
"Length: 294803701 (281M) [application/octet-stream]\n",
"Saving to: ‘stylegan3-t-metfacesu-1024x1024.pkl’\n",
"\n",
"stylegan3-t-metface 100%[===================>] 281.15M 44.7MB/s in 6.5s \n",
"\n",
"2021-10-12 04:18:01 (43.3 MB/s) - ‘stylegan3-t-metfacesu-1024x1024.pkl’ saved [294803701/294803701]\n",
"\n"
]
}
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "H4lWqDNdlIeU"
},
"source": [
"## Generate\n",
"\n",
"To find a good seed."
]
},
{
"cell_type": "code",
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "xWuRlgusuXi7",
"outputId": "f2615345-60c2-47ec-bb62-1b4ed08cd806"
},
"source": [
"!python --version"
],
"execution_count": null,
"outputs": [
{
"output_type": "stream",
"name": "stdout",
"text": [
"Python 3.7.12\n"
]
}
]
},
{
"cell_type": "code",
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "ea9v8IoklLxj",
"outputId": "dea9ac0b-cbd6-4ddf-d308-deea16d1db04"
},
"source": [
"!python gen_images.py --help"
],
"execution_count": null,
"outputs": [
{
"output_type": "stream",
"name": "stdout",
"text": [
" File \"gen_images.py\", line 34\n",
" if m := range_re.match(p):\n",
" ^\n",
"SyntaxError: invalid syntax\n"
]
}
]
}
]
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment