# 假设字符串型的标签存放在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里