INNER CODE UNIT · Python

exclusion_mask

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

    exclusion_mask = np.isin(seg_pred, identity_ids)
    save_mask(exclusion_mask, output_dir, "07_exclusion_mask_base")

    # Handle hands and feet
    if body_coverage in ("full", "upper"):
        hands_mask = seg_pred == labels_ids_dict["hands"]
        exclusion_mask = exclusion_mask | hands_mask

    if body_coverage in ("full", "lower"):
        feet_mask = seg_pred == labels_ids_dict["feet"]
        exclusion_mask = exclusion_mask | feet_mask

    save_mask(exclusion_mask, output_dir, "08_exclusion_mask_final")

    # Final mask
    final_mask = buffer_mask | (ca_mask_dilated & ~exclusion_mask)
    save_mask(final_mask, output_dir, "09_final_mask")

View source record →