nnie输入截断

海思

希望通过nnie对噪声图像经行去噪，由于送入nnie之前需要对图像经行预处理，图像实际的值范围超过了U8表示的0-255区间，因此现在用s32格式将预处理的图片送入nnie，但是测下来发现送入到nnie的图片会被截到（-1，1）之间，请问一下有哪位朋友知道是什么原因导致输入被截断的或者怎样送入范围不受限的图片。

我来回答

回答1个

时间排序

认可量排序

硅天下科技

1个粉丝

问答

专栏

资料

硅天下科技 2020-08-22 17:09:41

认可0 回复引用举报

海思提供的Sample的网络文件（即.wk文件）仅支持bgr格式图片，该wk文件对应的prototxt文件中设置了输入图片的颜色空间为BGR格式，每次放入测试图片时都要转为bgr格式稍显麻烦，其实还有一种方法直接读入JPEG格式图片，方法就是调用VDEC模块，而VDEC模块解码出来的是YUV颜色空间图像，因此还需要将.wk文件的颜色空间转为YVU格式，

使用VDEC模块需要绑定VPSS模块，使用VPSS模块还有个好处就是可以对图片进行缩放，VPSS模块转换为2路输出，一路将图片或视频转换为网络模型所需要的分辨率，再经过NNIE处理；另一路转换为输出所需要的分辨率如1920x1080；NNIE处理的结果经过VGS模块叠加在1920x1080分辨率图片/文件上，再经过VO输出。