search

keras-team / keras

Deep Learning for humans
http://keras.io/
Apache License 2.0
61.6k stars 19.41k forks source link

no ssim metrics for images #18428

Open anas-rz opened 1 year ago

anas-rz commented 1 year ago

Some networks e.g. require computing SSIM between images. Keras core doesn't have cross-backend ops for this. Can I open a PR for this?

fchollet commented 1 year ago

This could definitely be a useful addition to keras.ops.image. Do standard implementations exist in torch and JAX?

anas-rz commented 1 year ago

Torch and Jax don't have standard implementation. Can we implement it from scratch using ops?

fchollet commented 1 year ago

Maybe? What would that look like?

anas-rz commented 1 year ago

Will look like the tensorflow implementation where it uses depthwise convolution.

fchollet commented 1 year ago

It looks like it's implementable via a simple combination of a small number of existing backend ops. Sounds good to me!