Clamp YUV->RGB out of gamut

chaiNNer-org · Oct 14, 2023 · 288bb66 · 288bb66
1 parent 8daea7f
commit 288bb66
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backend/src/nodes/impl/color/convert_data.py b/backend/src/nodes/impl/color/convert_data.py
@@ -257,7 +257,7 @@ def __lch_to_lab(img: np.ndarray) -> np.ndarray:
     ),
     Conversion(
         direction=(YUV, RGB),
-        convert=lambda i: cv2.cvtColor(__rev3(i), cv2.COLOR_YUV2BGR),
+        convert=lambda i: np.clip(cv2.cvtColor(__rev3(i), cv2.COLOR_YUV2BGR), 0, 1),
         cost=__CHROMA_LOST,
     ),
     # HSV/HSL