change estimator settings

Author: EmoTim

age_estimator.py

@@ -6,10 +6,10 @@
 
 class AgeEstimator:
     def __init__(
-        self, ctx_id: int = 0, det_size: tuple[int, int] = (1024, 1024)
+        self, ctx_id: int = -1, det_size: tuple[int, int] = (640, 640)
     ) -> None:
         self.app = FaceAnalysis(name="buffalo_l")  # SOTA Model including age
-        self.app.prepare(ctx_id=ctx_id, det_size=det_size)  # 0 = gpu
+        self.app.prepare(ctx_id=ctx_id, det_size=det_size)  # -1 = CPU, 0+ = GPU
 
     def estimate_age(self, pil_img: Image.Image) -> float | None:
         """

generate.py

@@ -229,7 +229,7 @@ def generate_images(
     age_estimator = None
     if weight_vector is not None:
         print("Initializing age estimator...")
-        age_estimator = AgeEstimator(ctx_id=0, det_size=(1024, 1024))
+        age_estimator = AgeEstimator(ctx_id=-1, det_size=(640, 640))
 
     # Generate images.
     all_images = []  # Store images for composite: list of lists (one per seed)