mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-08-22 14:24:40 -07:00
Merge remote-tracking branch 'origin/master'
This commit is contained in:
Brigham Lysenko
commit
619f5515c7
1 changed files with 14 additions and 144 deletions
158
colab.ipynb
158
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": []
|
||||
}
|
||||
]
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue