利用多进程 处理图片数据 简单易懂

import cv2
from multiprocessing import Pool
import os

# 目的是都要对一批图片进行resize操作。

# 这里定义一个函数,只是一个简单的resize操作
def resize(image_path):
	print("pid: ", os.getpid())
	# name = os.path.basename(image_path)
	image = cv2.imread(image_path)
	print(type(image))
	image = cv2.resize(image, (400, 400), interpolation=cv2.INTER_CUBIC)
	cv2.imwrite(image_path, image)

if __name__ == '__main__':
	root_dir = "/Users/mayilong/Desktop/images"
	namelist = [os.path.join(root_dir, image_name) for image_name in os.listdir(root_dir)]
	print(namelist)
	# 创建进程池
	pool = Pool()
	# 开始进行处理,等待结果完成
	pool.map(resize, namelist)
发布了227 篇原创文章 · 获赞 13 · 访问量 8万+

猜你喜欢

转载自blog.csdn.net/weixin_36149892/article/details/96583762