海思AI芯片(Hi35XX): 图像jpg转.bgr

前言

把一张jpg图片转成bgr(注意顺序是bgr)二进制文件，最近在弄华为Hi3516DV300开发板上的一些深度学习的样例程序，换言之，35xx nnie SDK只支持BGR以及yuv格式的数据输入。那么当手头刚好只有jpeg/png等图片文件时，怎么来进行目标识别呢，一般的思路是用opencv来读取图片文件，然后再转换成BGR/YUV等格式，最后扔到NNIE模块来进行深度学习推理。

本篇讲解jpg转bgr，也就是它的输入文件不是普通的jpg，而是bgr格式的二进制文件，还别说，这样确实是大大节省了空间。一张3.8M的测试文件(test.jpg)，转成二进制后为276K。本篇使用python来实现，网上大部分都是c/c++实现的，对于一个python用习惯并且用来验证结果，我觉得python转化更为习惯吧。

本代码仅面向学习海思35xx的各位。转化完不是标准bgr格式，由于Hi35xx的bgr分布不是标准格式，所以写了一点代码进行bgr与jpg格式图片互相转化,并逆向思维进行测试转化后的bgr。

代码部分

# -*- coding: UTF-8 -*-
# author: AnnSun
# date :2020.05.12import os
import cv2
from numpy import *
import numpy as npclass JPG2BGR_Solver(object):def __init__(self):self.img_size = 416  # save bgr size# jpj2bgr  converbgr=true 参数设置self.imgpath_jpg = r"pose_368.jpg"self.saveimg_bgr = r"pose_368.bgr"# testbgr  converbgr=false 参数设置self.jpeg_path = r"./BGR_img/double_roipooling_224_224.jpg"self.path = "./BGR_img/24_608x608.bgr""""海思nnie模型需要输入bgr 格式的图片，这个python脚本可以把jpg格式的图片转换成.bgr格式的图片"""def jpg2bgr(self):save_img_size = self.img_sizeimgpath = self.imgpathimg = cv2.imread(imgpath)if img is None:print("img is none")else:img = cv2.resize(img,(save_img_size,save_img_size))(B, G, R) = cv2.split(img)with open(saveimg,'wb')as fp:for i in range(save_img_size):for j in range(save_img_size):fp.write(B[i, j])print(B[i,j])for i in range(save_img_size):for j in range(save_img_size):fp.write(G[i, j])for i in range(save_img_size):for j in range(save_img_size):fp.write(R[i, j])print("save success")"""查看bgr文件内容并显示为图片"""def test_Hi_bgr(self):jpeg_path = self.jpeg_pathpath = self.pathimgsize = self.img_sizef = open(path, 'rb')src = cv2.imread(jpeg_path)src = cv2.resize(src, (imgsize, imgsize))print(src.shape)h = src.shape[0]w = src.shape[1]c = src.shape[2]print(f.name)(B, G, R) = cv2.split(src)data = f.read(imgsize*imgsize*3)for j in range(imgsize):for i in range(imgsize):B[j, i] = data[j*imgsize + i]G[j, i] = data[j*imgsize + i + imgsize*imgsize]R[j, i] = data[j*imgsize + i + imgsize*imgsize*2]newimg = cv2.merge([B, G, R])cv2.imshow("new", newimg)f.close()cv2.waitKey(0)if __name__ == '__main__':converbgr = truesolverObj = JPG2BGR_Solver()if (converbgr == true):solverObj.jpg2bgr()else:solverObj.test_Hi_bgr()

海思AI芯片(Hi35XX): 图像jpg转.bgr相关推荐

海康摄像头海思AI芯片(Hi35XX): 图像jpg转.bgr
前言把一张jpg图片转成bgr(注意顺序是bgr)二进制文件,最近在弄华为Hi3516DV300开发板上的一些深度学习的样例程序,换言之,35xx nnie SDK只支持BGR以及yuv格式的数据输 ...
海思AI芯片(Hi35XX): 图像jpg转.bgr升级版
https://blog.csdn.net/smf0504/article/details/106068342 我把上面这个老哥的代码改进了一下,小改进,备份下,免得丢了主要是这个,self.img ...
海思AI芯片(Hi3519A/3559A)方案学习（一）资料以及术语介绍
芯片系列介绍海思AI芯片Hi35xx系列具体包括的型号Hi3559A V100, Hi3559C V100, Hi3519A V100, Hi3516C V500以及Hi3516D V300.下面要 ...
华为海思AI芯片HI3516DV300上SSD代码的移植和优化
前一段时间一直在做海思AI芯片 HI3516DV300上SSD代码的优化工作,这里总结一下优化的一些心得体会.之所以要对海思AI芯片 HI3516DV300上SSD代码做优化,是因为SSD有一部分是运 ...
海思AI芯片(Hi3519A/3559A)方案学习（十四）JPEG图片转换成bgr文件
原文:https://blog.csdn.net/avideointerfaces/article/details/89931156 前言在系列文章海思AI芯片(Hi3519A/3559A)方案学习 ...
海思AI芯片(Hi3519A/3559A)方案学习（十七）开发板上运行yolo3模型的代码分析
前言前面的博客系列已经介绍了如何将caffemodel转换成wk文件,如何将jpg文件转成bgr格式数据以及如何在PC上仿真模型推理等,基于这些基础,本文来结合代码分析如何在板子上推理yolov3 ...
海思AI芯片(Hi3519A/3559A)方案学习（四）如何在3519A板子上运行sample code
sample code编译先解压缩SDK(Hi3519AV100_SDK_V2.0.1.0.tgz)得到Hi3519AV100_SDK_V2.0.1.0文件夹.其sample源代码藏在Hi3519A ...
【海思AI芯片Hi3559A】学习笔记(1)：开发环境搭建、安装SDK
芯片资料见Hi3559A的资料包主要参考文档:ReleaseDoc\zh\01.software\board\Hi3559A╱C V100 SDK 安装及升级使用说明.pdf 注意所有解压操作均要在 ...
海思AI芯片(Hi3519A/3559A)方案学习（二十六）VPSS不同channel的不同放缩能力
前言在用VPSS的channel1对vdec解码出来的图片进行放大时,比如说(260,260)放大到(928,608),遇到了类似下面的错误. [Func]:VPSS_DRV_CheckVpssCa ...

海思AI芯片(Hi35XX): 图像jpg转.bgr

海思AI芯片(Hi35XX): 图像jpg转.bgr相关推荐

最新文章

热门文章