Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Plans for implementing SSIM loss #1023

Closed
SomritaRPE opened this issue Mar 31, 2023 · 3 comments
Closed

Plans for implementing SSIM loss #1023

SomritaRPE opened this issue Mar 31, 2023 · 3 comments

Comments

@SomritaRPE
Copy link

Hi All

I am unable to find any loss function in Julia corresponding to SSIM .Will there be any upgradation regarding this?
I've used it many time is the TensorFlow platform but there's no equivalent (that I could find) in Julia Images .It is very important loss for image to image translations, thus it would be grate if it becomes available here.

Best regards

Somrita

@ashwani-rathee
Copy link
Member

Hey, I think you are mentioning this: https://github.com/JuliaImages/ImageQualityIndexes.jl/blob/master/src/ssim.jl

@ashwani-rathee
Copy link
Member

This is being worked on here: FluxML/Flux.jl#2178

@nikopj
Copy link

nikopj commented Jul 14, 2023

I've moved the work of the Flux pull request to a standalone repository here: https://github.com/nikopj/SSIMLoss.jl
SSIMLoss.jl is AD and GPU friendly + tested against ImageQualityIndexes.jl for correctness.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants