python-keras图像多标签分类模型训练

CSDN话题挑战赛第2期
参赛话题：学习笔记

最近在学习给图像打标签训练，记录一下自己整个过程。

在python中有数据包Tensorflow-keras可以进行模型训练

数据集可以是自己采集也可以下载网络上的数据集

  1.将image和其对应的labels进行处理 ，写入excel表

   2.获取所有图像进行读取，裁剪，转化为数组

# image_list 是图像列表
# data 是存放图像数组列表
# labels 是图像对应标签的列表
for imagePath in image_list:
	image = cv2.imread(imagePath)
	image = cv2.resize(image, (writh, height))
	image = img_to_array(image)
	data.append(image)
    label = image_dict[imagePath].split(',')
	labels.append(label)

   3.将data和labels列表转为数组

data = np.array(data, dtype="float") / 255.0
labels = np.array(labels)

   4.labels进行独热编码

mlb = MultiLabelBinarizer()
labels