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

Where the functions are? #251

Closed RizerKayseri closed 1 year ago

RizerKayseri commented 1 year ago

When i read the program written by others, some errors have occurred.

I cannot reference the following functions from compressai.ops import ste_round as ste_round from compressai.ans import BufferedRansEncoder, RansDecoder

ayuzy commented 6 months ago

I have encountered the same problem, may I ask how this is solved

YodaEmbedding commented 6 months ago

Change the line giving the ImportError to:

import compressai.ops.ops.quantize_ste as ste_round
RizerKayseri commented 6 months ago

Change the line giving the ImportError to:

import compressai.ops.ops.quantize_ste as ste_round

thank you very much

RizerKayseri commented 6 months ago

I have encountered the same problem, may I ask how this is solved

It is difficult to recall my method at that time; I suggest that you review another response. https://github.com/InterDigitalInc/CompressAI/issues/251#issuecomment-1958896803

dylan-d-hub commented 5 months ago

from compressai.ops import quantize_ste as ste_round