LiuLei95 / PyTorch-Learned-Image-Compression-with-GMM-and-Attention

This repo is implementation for Learned Image Compression with Discretized Gaussian Mixture Likelihoods and Attention Modules in pytorch.
Apache License 2.0
61 stars 2 forks source link

编码和解码 #9

Open Bai-shen opened 1 year ago

Bai-shen commented 1 year ago

很抱歉打扰了,想请教一下这个代码里,输入图像编码压缩与解码的流程在哪里

Bai-shen commented 1 year ago

clipped_recon_image, mse_loss, bpp_feature, bpp_z, bpp = net(input) 还有这里,这几个参数都代表了什么,求助QAQ