From ccae8536bf4ffcab7a6f546e5c77f8ea8a021452 Mon Sep 17 00:00:00 2001 From: Hawk Lee Date: Tue, 20 Jan 2026 20:57:16 +0800 Subject: [PATCH] fix: Revert crop to 512 for generator comp, update detector input to 800 --- libs/Ditto/core/atomic_components/source2info.py | 2 +- libs/Ditto/core/aux_models/insightface_det.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/Ditto/core/atomic_components/source2info.py b/libs/Ditto/core/atomic_components/source2info.py index 3322f54..1a64fb6 100644 --- a/libs/Ditto/core/atomic_components/source2info.py +++ b/libs/Ditto/core/atomic_components/source2info.py @@ -89,7 +89,7 @@ class Source2Info: ret_dct = crop_image( img, lmk203, - dsize=800, + dsize=512, scale=kwargs.get("crop_scale", 2.3), vx_ratio=kwargs.get("crop_vx_ratio", 0), vy_ratio=kwargs.get("crop_vy_ratio", -0.125), diff --git a/libs/Ditto/core/aux_models/insightface_det.py b/libs/Ditto/core/aux_models/insightface_det.py index 10a6774..756ae1f 100644 --- a/libs/Ditto/core/aux_models/insightface_det.py +++ b/libs/Ditto/core/aux_models/insightface_det.py @@ -71,7 +71,7 @@ class InsightFaceDet: self.nms_thresh = 0.4 self.det_thresh = 0.5 - self.input_size = (512, 512) + self.input_size = (800, 800) self.input_mean = 127.5 self.input_std = 128.0 self._anchor_ratio = 1.0