写代码的过程中,碰到了要获取本地文件夹中图像的宽高信息,由于文件夹中的图像数量较多,有两万多张图像。运行程序时,发现耗时严重,经过定位分析,发现是因为使用OpenCV来获取图像的宽高,代码如下:
img = cv2.imread(str(img_path)) # opencv 加载图像耗时大概 42 - 45 ms, 程序运行慢的原因出现在这里
img_h, img_w, _ = img.shape
使用OpenCV加载图像后,再通过shape属性获取图像宽高信息,平均耗时在42~45ms,加载两万张图像的耗时至少为42×20000/1000/60=14分钟。
后来,发现可以使用imagesize库来获取图像的宽高信息,代码如下:
img_w, img_h = imagesize.get(str(img_path)) # 耗时不到 1 ms
平均一张图像耗时不到1ms,两万张图像耗时最多1×20000/1000/60=0.33分钟,比之前使用OpenCV的方法快了近42倍。
imagesize是python的开源库,官方链接:https://github.com/shibukawa/imagesize_py。