Python数据处理 - 查看海量图片

       目标检测模型在测试集上做推理的时候,有时会是几十万张的规模,这样就没办法用文件资源管理器打开,会卡死,只能用程序一张张打开查看效果。

import matplotlib.pyplot as plt # plt 用于显示图片
import matplotlib.image as mpimg # mpimg 用于读取图片
import numpy as np
import time
import os

# backend = matplotlib.get_backend() # 后端内核
# print(backend) # 我的是 qt5agg

path = '/mnt/work/5voloy/runs/detect/exp17/'

plt.ion() #开启interactive mode

for filename in os.listdir(path):
	thePic = mpimg.imread(path + filename) # 此时 thePic 已被读取为 np.array 
	thePic.shape # (512, 512, 3)
	plt.figure().canvas.manager.window.move(900, 500) # 我的屏幕的中央坐标
	plt.imshow(thePic) # 显示图片
	plt.axis('on') # 是否显示坐标轴
	plt.draw()
	# time.sleep(3) # 用这个会黑屏
	plt.pause(3)# 间隔的秒数： 3s
	plt.close()

 

参考文章：

Python在plt.show()后关闭窗口，继续运行命令的方法_打酱猪的博客-CSDN博客_plt.close

python实现读取并显示图片的两种方法 - 蘭亭客 - 博客园