python按标签分类切分数据_#使用python处理分割数据集，并按比例随机划分训练集与测试集...

对分割数据集进行打label处理

一些分割数据集会给出类似的图片作为label

我们要将其对应的灰度值转化为标签

获取文件列表

def get_files(files_dir):

files_list = []

for file in os.listdir(files_dir):

files_list.append(file)

return files_list

根据标签进行转换

files_dir = './label/'

files_list = get_files(files_dir)

print(len(files_list))

for i in range(len(files_list)):

a = Image.open("./label/" + files_list[i])

#a.show()

a =np.array(a)

a[a == 0] = 0

a[a == 200] = 1

a[a == 204] = 1

a[a == 213] = 255

a[a == 209] = 1

a[a == 206] = 255

a[a == 207] = 255

a[a == 201] = 2

a[a == 203] = 2

a[a ==