You can see them define a custom color loss and apply it simultaneously with the regular diffusion loss. I've actually expanded this notebook to allow regional specification of the custom loss.
It's quite difficult to define a function that detects if an individual has 5 fingers or not. That's the real issue.
https://colab.research.google.com/drive/1dlgggNa5Mz8sEAGU0wF...
You can see them define a custom color loss and apply it simultaneously with the regular diffusion loss. I've actually expanded this notebook to allow regional specification of the custom loss.
It's quite difficult to define a function that detects if an individual has 5 fingers or not. That's the real issue.