From b85773c57dee7be903b48dfe820120100b3164eb Mon Sep 17 00:00:00 2001 From: MengAiDev <3463526515@qq.com> Date: Thu, 7 Aug 2025 16:29:23 +0800 Subject: [PATCH] fix(fluX image validation): convert RGB to BGR for OpenPose images - Add conversion from RGB to BGR for validation images in the fluX control example - Ensure compatibility with OpenPose processing which expects BGR format --- examples/flux-control/train_control_lora_flux.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/examples/flux-control/train_control_lora_flux.py b/examples/flux-control/train_control_lora_flux.py index 2990d5701a0d..9ee8202b1c00 100644 --- a/examples/flux-control/train_control_lora_flux.py +++ b/examples/flux-control/train_control_lora_flux.py @@ -125,6 +125,9 @@ def log_validation(flux_transformer, args, accelerator, weight_dtype, step, is_f for validation_prompt, validation_image in zip(validation_prompts, validation_images): validation_image = load_image(validation_image) + # Convert RGB to BGR for OpenPose validation images + validation_image_np = np.array(validation_image)[:, :, ::-1] + validation_image = Image.fromarray(validation_image_np) # maybe need to inference on 1024 to get a good image validation_image = validation_image.resize((args.resolution, args.resolution))