基于深度学习的衣物分类识别 yolov8图像分类之衣物分类 衣服颜色识别 虚拟穿衣数据集

张开发
2026/4/15 12:34:12 15 分钟阅读

分享文章

基于深度学习的衣物分类识别 yolov8图像分类之衣物分类 衣服颜色识别 虚拟穿衣数据集
文章目录数据集划分数据预处理下载结构数据集中的类别训练使用的模型YOLOv8s-cls演示颜色处理识别主色功能匹配颜色功能衣物分类数据集用于此训练的数据集可以通过以下链接在 我这上找到。数据集划分数据集训练集验证集测试集图片数量6037 张682 张742 张占比81%9%10%数据预处理自动方向已应用调整大小拉伸至 640x640下载结构文件夹结构数据集中的类别连衣裙帽子长袖外套裤子衬衫鞋子短裤裙子T恤训练使用的模型YOLOv8s-cls模型大小 (像素)top1 准确率top5 准确率CPU ONNX 速度 (ms)A100 TensorRT 速度 (ms)参数量 (M)FLOPs (B) at 640YOLOv8n-cls22469.088.312.90.312.74.3YOLOv8s-cls22473.891.723.40.356.413.5YOLOv8m-cls22476.893.585.40.6217.042.7YOLOv8l-cls22476.893.5163.00.8737.599.7YOLOv8x-cls22479.094.6232.01.0157.4154.8演示原始图像原始图像带预测的图像带预测的图像颜色处理识别主色功能recognize_predominant_color函数将打开 PNG 图像将像素转换为 RGBA 格式并遍历这些像素忽略透明度 (alpha) 值为 0 的像素。它会统计每种非透明颜色的频率并返回最主要的颜色。返回的主色为: (30, 45, 78)裤子的颜色也是裤子颜色匹配颜色功能matching_colors函数接受一个选定颜色和一组颜色列表作为输入此外还包括一个容差值。它将返回颜色列表中与选定颜色在指定容差范围内匹配的颜色。用法selected_color: 你希望找到匹配的 RGB 颜色元组。color_list: 一组 RGB 颜色元组。tolerance: 一个数值表示允许选定颜色与列表中颜色之间的最大差异超过该差异则不被视为匹配。

更多文章