Update SimSwap colab.ipynb

This commit is contained in:
NNNNAI
2021-07-03 23:09:04 +08:00
parent c9b15112ae
commit 875c48ce3e
+5 -5
View File
@@ -204,9 +204,9 @@
"output_type": "stream",
"text": [
" crop_224\t models\t\t test_one_image.py\n",
" data\t\t options\t\t test_video_swapmutil.py\n",
" data\t\t options\t\t test_video_swapmulti.py\n",
" demo_file\t output\t\t test_video_swapsingle.py\n",
" doc\t\t README.md\t\t test_wholeimage_swapmutil.py\n",
" doc\t\t README.md\t\t test_wholeimage_swapmulti.py\n",
" insightface_func 'SimSwap colab.ipynb' test_wholeimage_swapsingle.py\n",
" LICENSE\t simswaplogo\t\t util\n"
],
@@ -329,7 +329,7 @@
"from torchvision import transforms\n",
"from models.models import create_model\n",
"from options.test_options import TestOptions\n",
"from insightface_func.face_detect_crop_mutil import Face_detect_crop\n",
"from insightface_func.face_detect_crop_multi import Face_detect_crop\n",
"from util.videoswap import video_swap\n",
"from util.add_watermark import watermark_image"
],
@@ -387,7 +387,7 @@
"opt.parser.add_argument('-f') ## dummy arg to avoid bug\n",
"opt = opt.parse()\n",
"opt.pic_a_path = './demo_file/Iron_man.jpg' ## or replace it with image from your own google drive\n",
"opt.video_path = './demo_file/mutil_people_1080p.mp4' ## or replace it with video from your own google drive\n",
"opt.video_path = './demo_file/multi_people_1080p.mp4' ## or replace it with video from your own google drive\n",
"opt.output_path = './output/demo.mp4'\n",
"opt.temp_path = './tmp'\n",
"opt.Arc_path = './arcface_model/arcface_checkpoint.tar'\n",
@@ -489,7 +489,7 @@
"use_dropout: False\n",
"use_encoded_image: False\n",
"verbose: False\n",
"video_path: ./demo_file/mutil_people_1080p.mp4\n",
"video_path: ./demo_file/multi_people_1080p.mp4\n",
"which_epoch: latest\n",
"-------------- End ----------------\n",
"input mean and std: 127.5 127.5\n",