Tools

Dataset Tools

A tool that help to generate captions with BLIP, BLIP2 or WD14 and mask for masked training using ClipSeg or Rembg. For caption generation, if you set an initial caption, it starts generating from that text instead of an empty string (BLIP and BLIP2 only).

Mask Generation

There are a few different tools to automatically generate a mask, depending on your image type, included in OneTrainer under the dataset tools button. With the batch generate mask tool, you can use ClipSeg, Rembg, Rembg-human and Hex Color.

With ClipSeg, you can use prompts such as "a woman" or "face of a woman" or "face and hair of a woman" to have the model create a mask outside of the areas you specify.

With the manual paint features, you can mask an area off and then use the fill option to fill in the remainder instead of trying to use a brush.

Do not forget to press enter after you are done manually editing your mask!

Convert Model Tools

Convert between different model formats.