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.19k stars 232 forks source link

GMM moduls is not used in cheng2020-anchor? #15

Closed ywz978020607 closed 3 years ago

ywz978020607 commented 3 years ago

Excuse me, I saw the cheng2020-anchor code and found that the entropy model is used with a single Gaussion model, but in the author paper, the Gaussion Mixture Model (GMM) is used, could you fresh these code? Thanks~

jbegaint commented 3 years ago

Hi, yes we have only implemented the "anchor" model and the "anchor" model with the attention layers for now. We might implement the full version with the GMM based entropy model later in the future.

ywz978020607 commented 3 years ago

I have achieved these part, including GMM model with Range Coder. https://www.cnblogs.com/sharklet/p/12809114.html