LiuLei95 / PyTorch-Learned-Image-Compression-with-GMM-and-Attention

This repo is implementation for Learned Image Compression with Discretized Gaussian Mixture Likelihoods and Attention Modules in pytorch.
Apache License 2.0
65 stars 3 forks source link

Pretrained models for MS-SSIM #5

Open Yuval-H opened 2 years ago

Yuval-H commented 2 years ago

Thank you for the great code!

Would you happen to have any pretrained models weights trained with MS-SSIM loss?

LiuLei95 commented 2 years ago

We don't have any pretrained models trained with MS-SSIM loss.