从训练数据生成器获取模型训练所需的标签字典(key: 类别名,value: 对应的数值编码)

labels = {value: key for key, value in train_generator.class_indices.items()}

输出标签字典和类别之间的映射关系

print('训练和验证数据集中的类别标签映射关系\n') for key, value in labels.items(): print(f'{key} : {value}')

从训练数据生成器获取模型训练标签字典

原文地址: http://www.cveoy.top/t/topic/pggd 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录