OneTrainer icon indicating copy to clipboard operation
OneTrainer copied to clipboard

[Feat]: per-concept (or image) masking

Open yggdrasil75 opened this issue 10 months ago • 1 comments

Describe your use-case.

I am training a lora on some characters. some of the characters have more interaction with the background of the image (ie: halfway behind a door) and some have less (ie: leaning on a table) I would like to experiment with setting the % masked rate for the images separately

What would you like to see as a solution?

an option is set the mask probability on the concept another option is setting it in the filename. for instance: -masklabel..png (12345-masklabel.0.9.png maybe) next option is folder. ie: \0.9<files> for 0.9 probability

Have you considered alternatives? List them here.

an alternative I may use is just duplicating the images, sometimes with a mask, sometimes without. this will work, but will be highly impractical done manually. I may do it with a short script, but that still feels like a bad idea. on windows it will mean much greater space used unless its done via hardlinks.

yggdrasil75 avatar Apr 05 '24 20:04 yggdrasil75