将数据集 标签 转化为 整型

# 假设字符串型的标签存放在img_label这个list中

label_set = set(img_label)    
# 将img_label转化成集合的形式，集合中不会有重复的元素

img_label_ = sorted(list((label_set)))  
# 这里再把集合转化成list，
# 这里原始标签对应于哪一个整数并不重要
# 但是依然要排序，否则每次生成的img_label_不一样

dict = {}   
# 创建一个字典，将字符标签和整型标签对应起来

for i in range(len(label_set)):
    dict[img_label_[i]] = i

label = [dict[img_label[int(i)]] for i in img_label]
# 将img_label中的标签转化成对应的整型标签，存放到label这个list里