def __init__(self, method): if not isinstance(method, str): raise ValueError(f'Invalid method type "{type(method)}"') from compressai import available_entropy_coders if method not in available_entropy_coders(): methods = ", ".join(available_entropy_coders()) raise ValueError(f'Unknown entropy coder "{method}"' f" (available: {methods})") if method == "ans": from compressai import ans encoder = ans.RansEncoder() decoder = ans.RansDecoder() elif method == "rangecoder": import range_coder encoder = range_coder.RangeEncoder() decoder = range_coder.RangeDecoder() self.name = method self._encoder = encoder self._decoder = decoder
def __init__(self, method): if not isinstance(method, str): raise ValueError(f'Invalid method type "{type(method)}"') from compressai import available_entropy_coders if method not in available_entropy_coders(): methods = ', '.join(available_entropy_coders()) raise ValueError(f'Unknown entropy coder "{method}"' f' (available: {methods})') if method == 'ans': from compressai import ans # pylint: disable=E0611 encoder = ans.RansEncoder() decoder = ans.RansDecoder() elif method == 'rangecoder': import range_coder # pylint: disable=E0401 encoder = range_coder.RangeEncoder() decoder = range_coder.RangeDecoder() self._encoder = encoder self._decoder = decoder
def apply_range_decoder(seq_data_len, decodepath, args, config): resolution = config['resolution'] prob = np.load('data_info/distribution_info_{}.npy'.format(args.model_num)) # Avoid zero prob modified_freq = prob * resolution + 1 modified_prob = modified_freq / np.sum(modified_freq) # print(modified_prob) cum_freq = range_coder.prob_to_cum_freq(modified_prob, resolution=resolution) # print('-----') # print(cum_freq) # cum_freq = [0] + [i for i in range(1, 256 + 1)] range_decoder = range_coder.RangeDecoder(decodepath) # Whether cum_freq resolution influences performance ? seq_data = range_decoder.decode(seq_data_len, cum_freq) return seq_data