INNER CODE UNIT · Python

selected_labels_mask

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

    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")

    return result


def main():
    parser = argparse.ArgumentParser(
        description="Debug mask creation pipeline - visualizes all intermediate masks",
        formatter_class=argparse.RawDescriptionHelpFormatter,
        epilog="""
Example:
    python scripts/debug_masks.py

View source record →