jroessler/flux-kontext-segmentation-hat

Hat-LoRA for Segmenting Hat Regions in Fashion Images

Model Overview

Hat-LoRA is a lightweight LoRA adapter designed for the FLUX.1 Kontext generative model, enabling automatic segmentation of a "hat" region on fashion images—even when hats are absent from the original image. This LoRA is specifically intended for virtual try-on and inpainting workflows, where constructing a precise mask for new or modified clothing items is critical.

Intended Use

• Automatically generating a segmentation mask for a "hat" area.
• Facilitating downstream virtual try-on, inpainting, and style transfer tasks by providing a consistent mask for input images. Prompts used:
• Caption: put a green mask for a hat on the person

Acknowledgments

Based on the FLUX.1 Kontext generative model, trained with Ostris’ AI Toolkit. See video instructions.