备注:在python中使用opencv好像无法打开归一化图片,故推荐PIL中Image方法。
import os
from PIL import Image
img_path = 'C:/Users/**/Desktop/test_result/artery_test_results'
save_path = 'C:/Users/**/Desktop/test_result/artery_test_process'
img_list = os.listdir(img_path)
for img in img_list:
image = Image.open(os.path.join(img_path,img))
image = np.array(image) #将图像转换为数组
image[image>=0.3] = 255
image[image<0.3 ] = 0
image = np.array(image,np.uint8) #转换为整形
cv2.imwrite(os.path.join(save_path,img),image)