haakohu's picture
initial
5d756f1
raw
history blame contribute delete
534 Bytes
from dp2.anonymizer import Anonymizer
from dp2.detection.face_detector import FaceDetector
from ..defaults import common
from tops.config import LazyCall as L
detector = L(FaceDetector)(
face_detector_cfg=dict(name="DSFDDetector", clip_boxes=True),
face_post_process_cfg=dict(target_imsize=(256, 256), fdf128_expand=False),
score_threshold=0.3,
cache_directory=common.output_dir.joinpath("face_detection_cache"),
)
anonymizer = L(Anonymizer)(
detector="${detector}",
face_G_cfg="configs/fdf/stylegan.py",
)