【Atlas 200 DK使用FAQ】 NV21格式数据如何转换成NV12格式直接输入网络

举报
山人_mhc 发表于 2020/03/30 17:38:33 2020/03/30
【摘要】 Atlas 200 DK使用FAQ

图片处理相关案例

 

问题描述:

DVPP JPG解码生成YUV420SP的NV21格式的数据,但网络输入要求为YUV420SP的NV12格式的数据,应该如何进行转换?

 

问题分析:

AIPP支持输入YUV420SP的NV21格式的图片,若需要将NV21转换成NV12,可以通过AIPP配置文件中的rbuv_swap_switch配置项实现通道的交换。

将rbuv_swap_switch : false修改为rbuv_swap_switch : true即可。

图片19.png

 

解决方法:

AIPP在模型转换时进行,当前Mind Studio模型转换界面中的AIPP配置支持的输入图片格式为YUV420SP_u8(NV12)和RGB888_U8(RGB),如下图所示,不支持NV21的输入格式。

图片20.png

如果需要支持NV21或者BGR,需要通过命令行方式进行模型转换。

 

步骤1: 为了降低命令行方式下AIPP配置文件的配置复杂度,从Mind Studio中将AIPP的配置文件保存下来。通过Mind Studio操作界面进行模型转换,在convertModel.log日志里面会有aipp配置的打印,如下图所示,将红框中的打印信息拷贝到新文件中,例如命名为aipp.cfg。

图片21.png

将保存下来的aipp.cfg文件中的rbuv_swap_switch:false修改为rbuv_swap_switch:true。

 

步骤2:使用命令行方式进行模型转换。

例如:

omg   --model=resnet18.prototxt --weight=ResNet-18-model.caffemodel --framework=0 --output=resnet18  --insert_op_conf=aipp.cfg

omg工具的更多参数含义请参考omg  --help。

其中参数--output就是输出离线模型的名字,输出为resnet.om。

最后把om文件拷贝到自己需要运行的位置即可。

----结束

 

 

每日推送Ascend 310常见问题及处理方法,更多昇腾技术干货,尽在昇腾社区等你,敬请持续关注

 

 

【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。