Skip to content

Instantly share code, notes, and snippets.

@ry

ry/blur_loss.py Secret

Created January 4, 2016 18:44
Show Gist options
  • Star 4 You must be signed in to star a gist
  • Fork 4 You must be signed in to fork a gist
  • Save ry/148158137a1346eaea8e to your computer and use it in GitHub Desktop.
Save ry/148158137a1346eaea8e to your computer and use it in GitHub Desktop.
Download ZIP
Raw
blur_loss.py
def blur_uv_loss(rgb, inferred_rgb):
uv = rgb2uv(rgb)
uv_blur0 = rgb2uv(blur(rgb, 3))
uv_blur1 = rgb2uv(blur(rgb, 5))
inferred_uv = rgb2uv(inferred_rgb)
inferred_uv_blur0 = rgb2uv(blur(inferred_rgb, 3))
inferred_uv_blur1 = rgb2uv(blur(inferred_rgb, 5))
return ( dist(inferred_uv, uv) +
dist(inferred_uv_blur0 , uv_blur0) +
dist(inferred_uv_blur1, uv_blur1) ) / 3
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment