Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

不同的压缩率需要重新训练吗? #3

Open
Liu-MeiLi1 opened this issue Nov 16, 2022 · 3 comments
Open

不同的压缩率需要重新训练吗? #3

Liu-MeiLi1 opened this issue Nov 16, 2022 · 3 comments

Comments

@Liu-MeiLi1
Copy link

您好,如题, 看了文章之后没有找到答案,试跑了一下您的程序,发现按照默认值压缩率0.5训练之后的模型,简单改写了测试程序,加载模型更换压缩率会报尺寸不匹配的错误,不知道是不是改写的测试程序有误,用plt简单展示了一下output发现失去了原本的颜色,不知道您是怎么测试的,想请教一下您,谢谢!
image
image

image

@Liu-MeiLi1
Copy link
Author

啊 是两个问题

  1. 我output的维度是(1,1,480,360)不是3通道的彩色图,不知道题主知不知道是哪里我写错了
  2. 更换压缩率sensing-rate 会显示尺寸不匹配的问题,是不是需要重新训练模型
    谢谢作者!

@rkteddy
Copy link
Owner

rkteddy commented Nov 16, 2022

  1. 关于输入输出维度的问题:这份代码是先将RGB转换为YCbCr颜色空间,并取Y通道作为最终输入,也就是说,输入和输出默认都是单通道。如果需要三通道的话,得同时改写模型和训练代码。
  2. 不同压缩率的模型是不一样的。是的,需要重新训练。

@Liu-MeiLi1
Copy link
Author

明白了!谢谢您!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants