Losses
For losses to be used, custom_loss must be specified for that particular class. Currently there can be a total of four classes, saturation, smoothness, symmetry, and palette. To enable all losses use this syntax: separate loss names by commas, using a big string.
custom_loss="saturation, smoothness, symmetry, palette"

Saturation Loss

custom_loss="saturation"

saturation_weight

Default: 1
Creates a loss based on how colorful the image is, encourages the image to become more colorful.
saturation

Smoothness Loss

custom_loss="smoothness"

smoothness_weight

Default: 1
smoothness

Symmetry Loss

custom_loss="symmetry"

saturation_weight

Default: 1
symemtry

Palette Loss

custom_loss="palette"

palette_weight

Default: 1
palette
Copy link
On this page
Saturation Loss
saturation_weight
Smoothness Loss
smoothness_weight
Symmetry Loss
saturation_weight
Palette Loss
palette_weight