INNER CODE UNIT · Python

create_garment_image_debug

fashn-AI/fashn-vton-1.5 · scripts/debug_masks.py:195

def create_garment_image_debug(
    img_np: np.ndarray,
    seg_pred: np.ndarray,
    labels_to_segment_indices: list,
    output_dir: str,
    mask_value: int = 127,
) -> np.ndarray:
    """Create garment image with debug visualization."""
    print("\n=== Creating Garment Image (with debug) ===")

    # Create mask for selected labels
    selected_labels_mask = np.isin(seg_pred, labels_to_segment_indices)
    save_mask(selected_labels_mask, output_dir, "garment_01_selected_labels_mask")
    save_mask(~selected_labels_mask, output_dir, "garment_02_mask_to_fill")

    result = img_np.copy()
    result[~selected_labels_mask] = mask_value
    save_image(result, output_dir, "garment_03_result")

View source record →