smy20011 / dreambooth-gui

MIT License
362 stars 18 forks source link

FEATURE REQUEST: Some QoL for training parameters, bug in output window #11

Open jtkelm2 opened 1 year ago

jtkelm2 commented 1 year ago

Allow the user to choose a folder containing pre-generated class regularization images in the same way they can choose a folder for the instance images. Upon selection of a folder, automatically apply the proper flags (use_prior_preservation, num_class_images, etc.).

Include some hardcoded text fields/drop-down menus for the most common flags.

Allow preprocessing of training image directories like resizing, center-cropping, and flipping as in AUTO's webui. Alternatively, just include a checkbox that indicates whether you want to train on a flip-augmented dataset.

For some reason, all the r's have been replaced with n's in the training output window: image Maybe a failed attempt at converting carriage returns?

smy20011 commented 1 year ago

For the output bug, It should be fixed in 1.4 :)