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.15k stars 228 forks source link

feat: GaussianMixtureConditional #239

Closed lin-toto closed 7 months ago

lin-toto commented 1 year ago

238, Gaussian Mixture Model implementation which is used in Cheng2020 paper.

Implementation details follow the official code but I changed CDF generation into a parallel algorithm so that it's easier to work with parallel context models.

Also modified rans_interface to allow a torch tensor to be CDF, or ~80% of execution time is wasted on type conversion.

fracape commented 11 months ago

sorry for the late response, I was busy with other things. Can you give me push access to your fork for a quick fix before merge

lin-toto commented 11 months ago

sorry for the late response, I was busy with other things. Can you give me push access to your fork for a quick fix before merge

Thanks, though the "allow edits by maintainers" option seems already on my side. I've also sent an invitation to my fork if this doesn't work.

fracape commented 11 months ago

Correct, I messed-up and created a new branch, thanks