项目需求,前端需要将图片转换成base64格式发给后端的jetson AGX上处理。因为jetson AGX CPU性能是比较羸弱,解析的时候,一张图片4032*3024竟然要380ms。为了提升性能,选择了turboJpeg来替换opencv原生的cv2.imdecode()和cv2.imwrite(). 更换后大概能有200ms。

替换cv2.imwrite()

原代码:

import cv2
cv2.imwrite('img.jpg',img)

替换为

from turbojpeg import TurboJPEG,TJPF_GRAY,TJSAMP_GRAY,TJFLAG_PROGRESSIVE,TJPF_BGR
jpeg = TurboJPEG(turboJpeg_lib_path) # 这里放libTurboJpeg.so的地址

# 将图片输出到path
open_file = open(path,'wb')
open_file.write( jpeg.encode(image,quality=100))
open_file.close()

之前完成时间在750ms左右,换成turboJpeg后在260ms左右。

替换cv2.imdecode()

原代码

imgData = base64.b64decode(data)
nparr = np.frombuffer(imgData, np.uint8)
dst_image = cv2.imdecode(nparr, cv2.IMREAD_COLOR)

替换后

from turbojpeg import TurboJPEG,TJPF_GRAY,TJSAMP_GRAY,TJFLAG_PROGRESSIVE,TJPF_BGR
jpeg = TurboJPEG(turboJpeg_lib_path) # 这里放libTurboJpeg.so的地址

mgData = base64.b64decode(self.picData)
dst_image = jpeg.decode(imgData,pixel_format=TJPF_BGR)

之前完成时间在380ms左右,换成turboJpeg后在200ms左右。

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐