| # coding: utf-8 | |
| """ | |
| parameters used for crop faces | |
| """ | |
| import os.path as osp | |
| from dataclasses import dataclass | |
| from typing import Union, List | |
| from .base_config import PrintableConfig | |
| # use repr from PrintableConfig | |
| class CropConfig(PrintableConfig): | |
| device_id: int = 0 # gpu device id | |
| flag_force_cpu: bool = False # force cpu inference, WIP | |
| ########## source image cropping option ########## | |
| dsize: int = 512 # crop size | |
| scale: float = 2.5 # scale factor | |
| vx_ratio: float = 0 # vx ratio | |
| vy_ratio: float = -0.125 # vy ratio +up, -down | |
| max_face_num: int = 0 # max face number, 0 mean no limit | |
| ########## driving video auto cropping option ########## | |
| scale_crop_video: float = 2.2 #2.0 # scale factor for cropping video | |
| vx_ratio_crop_video: float = 0. # adjust y offset | |
| vy_ratio_crop_video: float = -0.1 # adjust x offset | |
| direction: str = 'large-small' # direction of cropping | |