Wrong ms-ssim q4 pretrained model for cheng2020-anchor - Githubissues

InterDigitalInc / CompressAI

A PyTorch library and evaluation platform for end-to-end compression research

https://interdigitalinc.github.io/CompressAI/

BSD 3-Clause Clear License

1.21k stars 232 forks source link

Wrong ms-ssim q4 pretrained model for cheng2020-anchor #100

Open felixcheng97 opened 3 years ago

felixcheng97 commented 3 years ago

Thanks for your great work!

When I download the ms-ssim q4 pretrained model for cheng2020-anchor

https://github.com/InterDigitalInc/CompressAI/blob/278f45e6f444448d3c44d973e51f9723768afa18/compressai/zoo/image.py#L149

, it seems that the channel dimension of this checkpoint is 128, which should be 192 instead.

https://github.com/InterDigitalInc/CompressAI/blob/278f45e6f444448d3c44d973e51f9723768afa18/compressai/zoo/image.py#L215-L222

Could you help fix this pretrained model? Thank you very much.

fracape commented 2 years ago

Hi sorry for the late reply. Good eye, thanks. I'll try to find the time to correct this. From our plots in this repo and experience, that should not change the performance by much at this mid-range of bitrates.