在进行深度学习中,即使我们历经千辛万苦训练模型并使用模型进行预测,最后也可能因为输出文件类型的错误导致项目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
待补充。。。