mirror of
https://github.com/KwaiVGI/LivePortrait.git
synced 2024-12-22 12:22:38 +00:00
feat: add scripts for one click windows (#219)
* feat: windows one-click installation * chore: format --------- Co-authored-by: zzzweakman <1819489045@qq.com>
This commit is contained in:
parent
172b852127
commit
1f443b2344
4
.gitignore
vendored
4
.gitignore
vendored
@ -22,3 +22,7 @@ tmp/*
|
|||||||
.vscode/launch.json
|
.vscode/launch.json
|
||||||
**/*.DS_Store
|
**/*.DS_Store
|
||||||
gradio_temp/**
|
gradio_temp/**
|
||||||
|
|
||||||
|
# Windows dependencies
|
||||||
|
ffmpeg/
|
||||||
|
LivePortrait_env/
|
||||||
|
5
app.py
5
app.py
@ -32,11 +32,14 @@ def fast_check_ffmpeg():
|
|||||||
tyro.extras.set_accent_color("bright_cyan")
|
tyro.extras.set_accent_color("bright_cyan")
|
||||||
args = tyro.cli(ArgumentConfig)
|
args = tyro.cli(ArgumentConfig)
|
||||||
|
|
||||||
|
ffmpeg_dir = os.path.join(os.getcwd(), "ffmpeg")
|
||||||
|
if osp.exists(ffmpeg_dir):
|
||||||
|
os.environ["PATH"] += (os.pathsep + ffmpeg_dir)
|
||||||
|
|
||||||
if not fast_check_ffmpeg():
|
if not fast_check_ffmpeg():
|
||||||
raise ImportError(
|
raise ImportError(
|
||||||
"FFmpeg is not installed. Please install FFmpeg (including ffmpeg and ffprobe) before running this script. https://ffmpeg.org/download.html"
|
"FFmpeg is not installed. Please install FFmpeg (including ffmpeg and ffprobe) before running this script. https://ffmpeg.org/download.html"
|
||||||
)
|
)
|
||||||
|
|
||||||
# specify configs for inference
|
# specify configs for inference
|
||||||
inference_cfg = partial_fields(InferenceConfig, args.__dict__) # use attribute of args to initial InferenceConfig
|
inference_cfg = partial_fields(InferenceConfig, args.__dict__) # use attribute of args to initial InferenceConfig
|
||||||
crop_cfg = partial_fields(CropConfig, args.__dict__) # use attribute of args to initial CropConfig
|
crop_cfg = partial_fields(CropConfig, args.__dict__) # use attribute of args to initial CropConfig
|
||||||
|
@ -1,5 +1,6 @@
|
|||||||
# coding: utf-8
|
# coding: utf-8
|
||||||
|
|
||||||
|
import os
|
||||||
import os.path as osp
|
import os.path as osp
|
||||||
import tyro
|
import tyro
|
||||||
import subprocess
|
import subprocess
|
||||||
@ -33,6 +34,10 @@ def main():
|
|||||||
tyro.extras.set_accent_color("bright_cyan")
|
tyro.extras.set_accent_color("bright_cyan")
|
||||||
args = tyro.cli(ArgumentConfig)
|
args = tyro.cli(ArgumentConfig)
|
||||||
|
|
||||||
|
ffmpeg_dir = os.path.join(os.getcwd(), "ffmpeg")
|
||||||
|
if osp.exists(ffmpeg_dir):
|
||||||
|
os.environ["PATH"] += (os.pathsep + ffmpeg_dir)
|
||||||
|
|
||||||
if not fast_check_ffmpeg():
|
if not fast_check_ffmpeg():
|
||||||
raise ImportError(
|
raise ImportError(
|
||||||
"FFmpeg is not installed. Please install FFmpeg (including ffmpeg and ffprobe) before running this script. https://ffmpeg.org/download.html"
|
"FFmpeg is not installed. Please install FFmpeg (including ffmpeg and ffprobe) before running this script. https://ffmpeg.org/download.html"
|
||||||
|
18
src/utils/check_windows_port.py
Normal file
18
src/utils/check_windows_port.py
Normal file
@ -0,0 +1,18 @@
|
|||||||
|
import socket
|
||||||
|
import sys
|
||||||
|
|
||||||
|
if len(sys.argv) != 2:
|
||||||
|
print("Usage: python check_port.py <port>")
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
port = int(sys.argv[1])
|
||||||
|
|
||||||
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||||
|
sock.settimeout(1)
|
||||||
|
result = sock.connect_ex(('127.0.0.1', port))
|
||||||
|
|
||||||
|
if result == 0:
|
||||||
|
print("LISTENING")
|
||||||
|
else:
|
||||||
|
print("NOT LISTENING")
|
||||||
|
sock.close
|
Loading…
Reference in New Issue
Block a user