OpenSourceJoker/LivePortrait
3
0
Fork 0
mirror of https://github.com/KwaiVGI/LivePortrait.git synced 2025-02-06 02:38:12 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: ndarray with default_factory

Browse Source
This commit is contained in:
guojianzhu 2024-07-17 17:00:26 +08:00
parent 0f839844f6
commit bc4aaa44bb
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/config/inference_config.py
View File

@ -4,10 +4,9 @@
config dataclass used for inference config dataclass used for inference
""" """
import os.path as osp
import cv2 import cv2
from numpy import ndarray from numpy import ndarray
from dataclasses import dataclass from dataclasses import dataclass, field
from typing import Literal, Tuple from typing import Literal, Tuple
from .base_config import PrintableConfig, make_abs_path from .base_config import PrintableConfig, make_abs_path
@ -46,7 +45,7 @@ class InferenceConfig(PrintableConfig):
crf: int = 15 # crf for output video crf: int = 15 # crf for output video
output_fps: int = 25 # default output fps output_fps: int = 25 # default output fps
mask_crop: ndarray = cv2.imread(make_abs_path('../utils/resources/mask_template.png'), cv2.IMREAD_COLOR) mask_crop: ndarray = field(default_factory=lambda: cv2.imread(make_abs_path('../utils/resources/mask_template.png'), cv2.IMREAD_COLOR))
size_gif: int = 256 # default gif size, TO IMPLEMENT size_gif: int = 256 # default gif size, TO IMPLEMENT
source_max_dim: int = 1280 # the max dim of height and width of source image source_max_dim: int = 1280 # the max dim of height and width of source image
source_division: int = 2 # make sure the height and width of source image can be divided by this number source_division: int = 2 # make sure the height and width of source image can be divided by this number