search

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

where is the source code of self.entropy_bottleneck.compress()? #216

Open JF-Lee opened 1 year ago

JF-Lee commented 1 year ago

Thanks of your excellent work! Now I wanna to deploy FactorizedPrior on embedded device, and I found EntropyBottleneck in it, but I can not find .compress() and decompress(), so can anyone tell me about that? Thanks again!

YodaEmbedding commented 1 year ago

EntropyBottleneck's compress/decompress:

https://github.com/InterDigitalInc/CompressAI/blob/ee91d536bd934fc1f8b1532f78db4c94072ae26d/compressai/entropy_models/entropy_models.py#L524-L537

Or are you referring to a "frozen"/"serialized"/"exported" model not having access to these functions?

JF-Lee commented 1 year ago

EntropyBottleneck's compress/decompress:

https://github.com/InterDigitalInc/CompressAI/blob/ee91d536bd934fc1f8b1532f78db4c94072ae26d/compressai/entropy_models/entropy_models.py#L524-L537

Or are you referring to a "frozen"/"serialized"/"exported" model not having access to these functions?

=================================================================== Thanks for your reply! I want to access to these functions, but I can not find that anywhere, so can you tell me where I can find that?