项目需求,前端需要将图片转换成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左右。