将文件夹中图像文件批量转换为uint8类型并保存

在进行深度学习中，即使我们历经千辛万苦训练模型并使用模型进行预测，最后也可能因为输出文件类型的错误导致项目GG。

现在展示一段便捷的代码来进行多个类型的转换。

即使同样是一个npy文件，其中的数据类型也可能不同。下面演示使用numpy的packet来转换npy文件。

import numpy as np
import os

x_test = []
img_path = r'E:\data'
all_files = os.listdir(img_path)
files = [item for item in all_files if "seg" in item]
for (n, file_name) in enumerate(files):
    img = np.load(os.path.join(img_path, file_name))
    img = np.uint8(img)
    np.save(os.path.join(img_path, file_name), img)

使用os.listdir来获取目标路径的所有文件目录。

用enumerate函数搭配for循环来遍历所有文件。

使用np.uint8()来转换类型。

最后记得将文件存在原地址来覆盖原文件。

我们还可以使用其他函数来进行其他类型转换：

1. uint8转float32

除以255的目的是让数值保持在0-1这个区间内

import numpy as np
img = img.astype(np.float32) / 255
print(img.dtype)

2. uint8转bit

待补充。。。