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

Last modified 1yr ago

Losses

`Saturation Loss`

`saturation_weight`

`Smoothness Loss`

`smoothness_weight`

`Symmetry Loss`

`saturation_weight`

`Palette Loss`

`palette_weight`