Method __init__

scripts/preprocess.py:34–45

Initialize the AnalyzeFace class with the given device, config file, and checkpoint file. Parameters: device (Union[str, torch.device]): The device to run the models on ('cuda' or 'cpu'). config_file (str): Path to the mmpose model configuration file. checkp

(self, device: Union[str, torch.device], config_file: str, checkpoint_file: str)

class AnalyzeFace:
    def __init__(self, device: Union[str, torch.device], config_file: str, checkpoint_file: str):
        """
        Initialize the AnalyzeFace class with the given device, config file, and checkpoint file.

        Parameters:
        device (Union[str, torch.device]): The device to run the models on ('cuda' or 'cpu').
        config_file (str): Path to the mmpose model configuration file.
        checkpoint_file (str): Path to the mmpose model checkpoint file.
        """
        self.device = device
        self.dwpose = init_model(config_file, checkpoint_file, device=self.device)
        self.facedet = FaceAlignment(LandmarksType._2D, flip_input=False, device=self.device)

    def __call__(self, im: np.ndarray) -> Tuple[List[np.ndarray], np.ndarray]:
        """