diff --git a/colab.ipynb b/colab.ipynb index 871f710..87b8095 100644 --- a/colab.ipynb +++ b/colab.ipynb @@ -13,7 +13,8 @@ "WTuyUxgdLA13", "avAcSL_uvtq_", "f7GNQ7kZx7Ha" - ] + ], + "toc_visible": true }, "kernelspec": { "name": "python3", @@ -33,9 +34,12 @@ "\n", "This is an adapted version of the DFL for Google Colab.\n", "\n", - "Version 2.5\n", + "# Face Shift Lab branch updates\n", + "* Turn off random face flipping during training.\n", + "* Multiple new color transfer methods.\n", + "* Ping pong. Bounce BS to improve efficiency and learning (expect errors, being fixed)\n", "\n", - "# Overview\n", + "# Original Feature Overview\n", "* Extractor works in full functionality.\n", "* Training can work without preview.\n", "* Converter works in full functionality.\n", @@ -64,7 +68,7 @@ "metadata": { "id": "JG-f2WqT4fLK", "colab_type": "code", - "cellView": "both", + "cellView": "form", "colab": {} }, "source": [ @@ -116,11 +120,7 @@ "id": "z4w_sUzgOQmL", "colab_type": "code", "cellView": "form", - "outputId": "9bfe6ad8-9f60-4707-9874-c2567784e5bc", - "colab": { - "base_uri": "https://localhost:8080/", - "height": 236 - } + "colab": {} }, "source": [ "#@title Import from Drive\n", @@ -158,25 +158,7 @@ "\n" ], "execution_count": 0, - "outputs": [ - { - "output_type": "stream", - "text": [ - "Go to this URL in a browser: https://accounts.google.com/o/oauth2/auth?client_id=947318989803-6bn6qk8qdgf4n4g3pfee6491hc0brc4i.apps.googleusercontent.com&redirect_uri=urn%3Aietf%3Awg%3Aoauth%3A2.0%3Aoob&scope=email%20https%3A%2F%2Fwww.googleapis.com%2Fauth%2Fdocs.test%20https%3A%2F%2Fwww.googleapis.com%2Fauth%2Fdrive%20https%3A%2F%2Fwww.googleapis.com%2Fauth%2Fdrive.photos.readonly%20https%3A%2F%2Fwww.googleapis.com%2Fauth%2Fpeopleapi.readonly&response_type=code\n", - "\n", - "Enter your authorization code:\n", - "··········\n", - "Mounted at /content/drive\n", - "/content\n", - "Done!\n", - "Mounted at /content/drive\n", - "/content\n", - "replace workspace/data_src/aligned/02319.jpg? [y]es, [n]o, [A]ll, [N]one, [r]ename: N\n", - "Done!\n" - ], - "name": "stdout" - } - ] + "outputs": [] }, { "cell_type": "code", @@ -514,11 +496,8 @@ "metadata": { "id": "Z0Kya-PJLDhv", "colab_type": "code", - "cellView": "both", - "outputId": "db06ff5b-4aae-4d4b-bad0-9c980782ba3b", - "colab": { - "base_uri": "https://localhost:8080/" - } + "cellView": "form", + "colab": {} }, "source": [ "#@title Training\n", @@ -566,116 +545,7 @@ "!python $train_cmd" ], "execution_count": 0, - "outputs": [ - { - "output_type": "stream", - "text": [ - "/content\n", - "Drive already mounted at /content/drive; to attempt to forcibly remount, call drive.mount(\"/content/drive\", force_remount=True).\n", - "Creating workspace archive ...\n", - "Archive created!\n", - "Time to end session: 12 hours\n", - "Running trainer.\n", - "\n", - "Loading model...\n", - "Press enter in 2 seconds to override model settings.\n", - "Enable autobackup? (y/n ?:help skip:n) : /usr/lib/python3.6/multiprocessing/semaphore_tracker.py:143: UserWarning: semaphore_tracker: There appear to be 1 leaked semaphores to clean up at shutdown\n", - " len(cache))\n", - "\n", - "n\n", - "Write preview history? (y/n ?:help skip:n) : \n", - "n\n", - "Target iteration (skip:unlimited/default) : \n", - "0\n", - "Batch_size (?:help skip:16) : 10\n", - "Enable ping-pong? (y/n ?:help skip:False) : ?\n", - "Cycles batch size between 1 and chosen batch size, simulating super convergence\n", - "Enable ping-pong? (y/n ?:help skip:False) : y\n", - "Ping-pong iteration (skip:1000/default) : \n", - "1000\n", - "Feed faces to network sorted by yaw? (y/n ?:help skip:y): \n", - "y\n", - "Flip faces randomly? (y/n ?:help skip:y) : ?\n", - "Predicted face will look more naturally without this option, but src faceset should cover all face directions as dst faceset.\n", - "Flip faces randomly? (y/n ?:help skip:y) : n\n", - "Optimizer mode? ( 1,2,3 ?:help skip:1) : \n", - "1\n", - "Use pixel loss? (y/n, ?:help skip: n ) : \n", - "n\n", - "Face style power ( 0.0 .. 100.0 ?:help skip:0.00) : \n", - "0.0\n", - "Background style power ( 0.0 .. 100.0 ?:help skip:0.00) : \n", - "0.0\n", - "Apply random color transfer to src faceset? (0) None, (1) LCT, (2) RCT, (3) RCT-c, (4) RCT-p, (5) RCT-pc, (6) mRTC, (7) mRTC-c, (8) mRTC-p, (9) mRTC-pc ?:help skip:False) : 0\n", - "Enable gradient clipping? (y/n, ?:help skip:y) : \n", - "y\n", - "Using TensorFlow backend.\n", - "WARNING:tensorflow:From /usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/op_def_library.py:263: colocate_with (from tensorflow.python.framework.ops) is deprecated and will be removed in a future version.\n", - "Instructions for updating:\n", - "Colocations handled automatically by placer.\n", - "WARNING:tensorflow:From /usr/local/lib/python3.6/dist-packages/tensorflow/python/ops/math_ops.py:3066: to_int32 (from tensorflow.python.ops.math_ops) is deprecated and will be removed in a future version.\n", - "Instructions for updating:\n", - "Use tf.cast instead.\n", - "Loading: 100% 4895/4895 [00:08<00:00, 562.57it/s]\n", - "Sorting: 100% 64/64 [00:00<00:00, 624.66it/s]\n", - "Loading: 0it [00:00, ?it/s]\n", - "Sorting: 100% 64/64 [00:00<00:00, 586103.62it/s]\n", - "========= Model Summary ==========\n", - "== ==\n", - "== Model name: SAE ==\n", - "== ==\n", - "== Current iteration: 202787 ==\n", - "== ==\n", - "==------- Model Options --------==\n", - "== ==\n", - "== batch_size: 10 ==\n", - "== sort_by_yaw: True ==\n", - "== random_flip: False ==\n", - "== resolution: 192 ==\n", - "== face_type: f ==\n", - "== learn_mask: True ==\n", - "== optimizer_mode: 1 ==\n", - "== archi: df ==\n", - "== ae_dims: 512 ==\n", - "== e_ch_dims: 42 ==\n", - "== d_ch_dims: 21 ==\n", - "== multiscale_decoder: True ==\n", - "== ca_weights: True ==\n", - "== pixel_loss: False ==\n", - "== face_style_power: 0.0 ==\n", - "== bg_style_power: 0.0 ==\n", - "== apply_random_ct: 0 ==\n", - "== clipgrad: True ==\n", - "== batch_cap: 10 ==\n", - "== ping_pong: True ==\n", - "== paddle: ping ==\n", - "== ping_pong_iter: 1000 ==\n", - "== ==\n", - "==--------- Running On ---------==\n", - "== ==\n", - "== Device index: 0 ==\n", - "== Name: Tesla T4 ==\n", - "== VRAM: 15.00GB ==\n", - "== ==\n", - "==================================\n", - "Starting. Press \"Enter\" to stop training and save model.\n", - "Process Process-2:\n", - "Traceback (most recent call last):\n", - " File \"/usr/lib/python3.6/multiprocessing/process.py\", line 258, in _bootstrap\n", - " self.run()\n", - " File \"/usr/lib/python3.6/multiprocessing/process.py\", line 93, in run\n", - " self._target(*self._args, **self._kwargs)\n", - " File \"/content/DeepFaceLab/utils/iter_utils.py\", line 39, in process_func\n", - " gen_data = next (self.generator_func)\n", - " File \"/content/DeepFaceLab/samplelib/SampleGeneratorFace.py\", line 85, in batch_func\n", - " raise ValueError('Not enough training data. Gather more faces!')\n", - "ValueError: Not enough training data. Gather more faces!\n", - "/usr/lib/python3.6/multiprocessing/semaphore_tracker.py:143: UserWarning: semaphore_tracker: There appear to be 1 leaked semaphores to clean up at shutdown\n", - " len(cache))\n" - ], - "name": "stdout" - } - ] + "outputs": [] }, { "cell_type": "markdown", @@ -725,4 +595,4 @@ "outputs": [] } ] -} \ No newline at end of file +}