Yolo V3识别颜色及长短袖

 

一、实现功能：识别长短袖衣服及长短裤子的颜色（分类为黑、白、彩）

 

二、运行环境

1. Ubuntu16.04

2. TensorFlow-gpu 1.8.0

3. Keras 2.1.5

4. GPU rtx2060

三、创建数据集

1. 使用VOC2007数据集的文件结构：

　　文件结构如下图，可以自己创建，也可以下载VOC2007数据集后删除文件内容。

　

 

2. 标注图片：

　　使用LabelImg(https://github.com/tzutalin/labelImg)对训练图片进行标注生成标签，然后将所有图片放在 JPEGImages 文件夹下，将所有标签(.xml)放在 Annotations 文件夹下。

3. 划分数据集：

　　python test.py  70%为训练集(train.txt), 15%为测试集(test.txt), 15%为评估数据集(val.txt)。txt文件保存在 ImageSets/Main 下。

4. 转换标签格式：

　　python voc_annotation.py, 在执行之前，需要根据数据集的实际类别