Upload reactor_swapper.py

custom_nodes/comfyui-reactor-node/scripts/reactor_swapper.py

@@ -304,7 +304,9 @@ def run_hyperswap(session, source_face, target_face, target_img):
 
     # Вычисляем аффинную матрицу
     M = get_affine_transform(target_landmarks_5.astype(np.float32), std_landmarks_256)
+    M_inv = cv2.invertAffineTransform(M)  # Обратная для paste
     logger.debug("Affine Matrix M (used for cropping):\n%s", M)
+    logger.debug("Inverse Affine Matrix M_inv (used for paste):\n%s", M_inv)
 
     # Применяем аффинное преобразование с новой матрицей M
     crop = cv2.warpAffine(target_img, M, (256, 256), flags=cv2.INTER_CUBIC, borderMode=cv2.BORDER_REFLECT)
@@ -358,6 +360,7 @@ def run_hyperswap(session, source_face, target_face, target_img):
     # (ваш код без изменений, но без старой денормализации)
     output = output.transpose(1, 2, 0)  # CHW -> HWC
     output = output[:, :, ::-1]  # RGB -> BGR (Убедитесь, что это BGR, если вход был BGR)
+    logger.debug("Output after denormalization: Min: %s | Max: %s", output.min(), output.max())
     
     # Визуализация после денормализации
     #### Что проверить:
@@ -367,11 +370,8 @@ def run_hyperswap(session, source_face, target_face, target_img):
   # cv2.imshow("Output After Denormalization", output)
   # cv2.waitKey(1)
     
-    # 7. Возвращаем результат 256x256 И матрицу M
-    # Мы больше не вызываем warp_and_paste_face отсюда.
-    # return result
-    
-    return output, M  # Возвращаем лицо (256x256) и матрицу M    
+    # 7. Возвращаем результат 256x256 И обратную матрицу M_inv
+    return output, M_inv    
 
 def sort_by_order(face, order: str):
     if order == "left-right":