Example #1
0
 def get_converter(self):
     from converters import ConverterMasked
     return ConverterMasked(self.predictor_func,
                            predictor_input_size=128,
                            face_type=FaceType.FULL,
                            base_erode_mask_modifier=30,
                            base_blur_mask_modifier=0)
Example #2
0
    def get_converter(self):
        base_erode_mask_modifier = 30 if self.options[
            'face_type'] == 'f' else 100
        base_blur_mask_modifier = 0 if self.options['face_type'] == 'f' else 100

        default_erode_mask_modifier = 0
        default_blur_mask_modifier = 100 if (self.options['face_style_power'] or self.options['bg_style_power']) and \
                                                self.options['face_type'] == 'f' else 0

        face_type = FaceType.FULL if self.options[
            'face_type'] == 'f' else FaceType.HALF

        from converters import ConverterMasked
        return ConverterMasked(
            self.predictor_func,
            predictor_input_size=self.options['resolution'],
            predictor_masked=self.options['learn_mask'],
            face_type=face_type,
            default_mode=1 if self.options['apply_random_ct']
            or self.options['face_style_power']
            or self.options['bg_style_power'] else 4,
            base_erode_mask_modifier=base_erode_mask_modifier,
            base_blur_mask_modifier=base_blur_mask_modifier,
            default_erode_mask_modifier=default_erode_mask_modifier,
            default_blur_mask_modifier=default_blur_mask_modifier,
            clip_hborder_mask_per=0.0625 if
            (self.options['face_type'] == 'f') else 0)
Example #3
0
 def get_converter(self):
     from converters import ConverterMasked
     return ConverterMasked(self.predictor_func,
                            predictor_input_size=64, 
                            output_size=64, 
                            face_type=FaceType.HALF, 
                            base_erode_mask_modifier=100,
                            base_blur_mask_modifier=100)
Example #4
0
    def get_converter(self):
        base_erode_mask_modifier = 30 if self.options[
            'face_type'] == 'f' else 100
        base_blur_mask_modifier = 0 if self.options['face_type'] == 'f' else 100

        default_erode_mask_modifier = 0
        default_blur_mask_modifier = 0

        face_type = FaceType.FULL if self.options[
            'face_type'] == 'f' else FaceType.HALF

        from converters import ConverterMasked
        return ConverterMasked(
            self.predictor_func,
            predictor_input_size=self.options['resolution'],
            predictor_masked=False,
            face_type=face_type,
            default_mode=4,
            base_erode_mask_modifier=base_erode_mask_modifier,
            base_blur_mask_modifier=base_blur_mask_modifier,
            default_erode_mask_modifier=default_erode_mask_modifier,
            default_blur_mask_modifier=default_blur_mask_modifier,
            clip_hborder_mask_per=0,
            force_mask_mode=5)