关于导出YUV文件的一些问题

[Ansarua]

   您好，关于您对自己工作的开源我表示非常感谢。
   我的问题是，当我用utils/file_io.py文件中的import_yuv工具将yuv数据集导入到dataset中，发现仅能返回一帧的Y通道信息。通过only_y这个参数可以修改成分别返回Y,U,V通道的信息，但重点是，分别返回3个通道的信息。我需要的是能不能直接返回channel=3的一个完整的ndarray。
   代码链接如下：https://github.com/ryanxingql/stdf-pytorch/blob/master/dataset/mfqev2.py 190行，这个是import_yuv工具的应用环境
   https://github.com/ryanxingql/stdf-pytorch/blob/master/utils/file_io.py   8行，这个是import_yuv工具的存在位置。
   但我不知道应该如何修改。还希望您能给予一些建议，非常感谢。

[ryanxingql]

视频源是 YUV420P 格式，意味着 U、V 分辨率要比 Y 小。此时不可能组成一个 ndarray。
除非你将 U、V 升采样至与 Y 等大，然后才可以组成一个数组。

[Ansarua]

视频源是YUV420P格式，那如果我直接用444的方式对它进行输出，生成的YUV能组成一个ndarray吗？

[ryanxingql]

视频源是YUV420P格式，那如果我直接用444的方式对它进行输出，生成的YUV能组成一个ndarray吗？

如果是444，意味着分辨率相同，可以组成一个ndarray。但前提是你要得到444。你需要升采样UV或降采样Y。

[Ansarua]

好的，那您知道出现这种保存结果是什么原因吗？值的范围没调整好？还是什么？

[ryanxingql]

色彩通道可能错了。例如 RGB 存成了 BGR。

[Ansarua]

您好，我尝试着修改过了，可是颜色还是不对，请问还有别的原因会导致这个问题吗？

[ryanxingql]

可以看一下完整代码吗？从YUV到RGB的

[Ansarua]

这部分并没有代码。我是将444格式的YUV分量直接cat成3通道的ndarray作为网络输入的。后面我察觉到可能是没转成RGB的问题。

请问网上的这个公式是否可用？或者您有YUV转RGB的代码可以分享一下吗？非常感谢

[ryanxingql]

可以试试。效果都差不多。建议参考：https://en.wikipedia.org/wiki/YUV

[Ansarua]

不太行，改成RGB直接看不到图了。

[Ansarua]

代码是这样写的。