欢迎光临
我们一直在努力

带标注的家具识别数据集,可识别床,椅子,餐桌, 门,水槽,沙发,马桶,浴缸等12种,识别率87%,支持yolo,coco json,pascal voc xml格式的模型训练

带标注的家具识别数据集,可识别床,椅子,餐桌, 门,水槽,沙发,马桶,浴缸等,识别率87%,支持yolo,coco json,pascal voc xml格式的模型训练

模型训练指标参数:

模型训练图:

数据集拆分

训练集

3190图像

验证集

84图像

测试集

67图像

预处理

自动定向:应用

调整大小:拉伸到640×640

增强

每个训练样本的输出数量:5

90° 旋转:顺时针、逆时针

旋转角度:-15° 至 +15° 之间

灰度化:应用于 15% 的图像

模糊:最大 2.5 像素

噪声:最多 0.1% 的像素添加

噪声裁剪遮挡:3 个遮挡框,每个尺寸为 10%

数据集标签:

'achair', 'bathtub', 'bed', 'chair', 'ctable', 'door', 'dtable', 'sink', 'sofa', 'ssink', 'ssofa', 'toilet'

数据集图片和标注信息示例:

数据集下载:

yolo26:https://download.csdn.net/download/pbymw8iwm/92774769

yolo v12:https://download.csdn.net/download/pbymw8iwm/92774768

yolo v11:https://download.csdn.net/download/pbymw8iwm/92774766

yolo v9:https://download.csdn.net/download/pbymw8iwm/92774770

yolo v8:https://download.csdn.net/download/pbymw8iwm/92774765

yolo v7:https://download.csdn.net/download/pbymw8iwm/92774763

yolo v5:https://download.csdn.net/download/pbymw8iwm/92774767

yolo darknet:https://download.csdn.net/download/pbymw8iwm/92774771

coco json: https://download.csdn.net/download/pbymw8iwm/92774772

pascal voc xml:https://download.csdn.net/download/pbymw8iwm/92774764

 YOLO模型训练

下载数据集之后解压到当前文件夹,然后将 我的仓库 https://gitcode.com/pbymw8iwm/YOLOProject里的训练模型脚本复制到文件夹下,假设你使用的是yolov8来训练你就用 python train_yolov8.py

注意,请根据你的GPU能力来适当调整训练参数,比如训练batch,patience,workers,以及模型类型(如果你的GPU硬件条件限制,可以联系作者进行付费模型训练,部分模型只需要一杯奶茶钱)

模型下载:

https://download.csdn.net/download/pbymw8iwm/92774837

模型验证测试情况:

验证测试代码:

#需要安装pip install ultralytics
from ultralytics import YOLO
import cv2

# 加载训练好的 YOLO .pt 模型
model = YOLO('best.pt') # 替换为你实际的 .pt 模型文件路径

# 定义要测试的图片路径
image_path = './image.jpg' # 替换为你实际的图片文件路径

# 使用模型对图片进行预测
results = model(image_path)

# 获取预测结果
for result in results:
# 获取绘制了检测框的图片
annotated_image = result.plot()

# 显示图片
cv2.imshow("YOLOv Inference", annotated_image)

# 等待按键退出
cv2.waitKey(0)

# 关闭所有 OpenCV 窗口
cv2.destroyAllWindows()

推理结果:

{   "predictions": [     {       "x": 80.5,       "y": 495,       "width": 121,       "height": 88,       "confidence": 0.905,       "class": "sofa",       "class_id": 8,       "detection_id": "8398ebd1-f2e2-45c8-857c-fbd32a5025f6"     },     {       "x": 148.5,       "y": 205.5,       "width": 51,       "height": 87,       "confidence": 0.905,       "class": "ctable",       "class_id": 4,       "detection_id": "14974335-9c0e-43b6-9ff5-e7f6464c692f"     },     {       "x": 149.5,       "y": 59.5,       "width": 51,       "height": 85,       "confidence": 0.904,       "class": "ctable",       "class_id": 4,       "detection_id": "15696dd0-b737-4030-9294-bb573ebbd913"     },     {       "x": 331,       "y": 464.5,       "width": 84,       "height": 123,       "confidence": 0.9,       "class": "dtable",       "class_id": 6,       "detection_id": "23f9f47c-26a5-4283-bbb3-c26898326631"     },     {       "x": 507.5,       "y": 362,       "width": 89,       "height": 70,       "confidence": 0.9,       "class": "bathtub",       "class_id": 1,       "detection_id": "c63a8c50-c5fe-42fb-9a9c-293b2c4d4cd3"     },     {       "x": 70,       "y": 394.5,       "width": 60,       "height": 93,       "confidence": 0.895,       "class": "ctable",       "class_id": 4,       "detection_id": "01ce1cd4-1206-4550-bc9f-e51f6893ca40"     },     {       "x": 413,       "y": 509.5,       "width": 70,       "height": 65,       "confidence": 0.88,       "class": "sink",       "class_id": 7,       "detection_id": "ab339152-dac7-4a86-8af3-c4dc7caa506b"     },     {       "x": 353,       "y": 67,       "width": 88,       "height": 98,       "confidence": 0.879,       "class": "bed",       "class_id": 2,       "detection_id": "7f11e368-d765-48d8-8ab2-242afc98abc7"     },     {       "x": 282,       "y": 300.5,       "width": 60,       "height": 129,       "confidence": 0.877,       "class": "door",       "class_id": 5,       "detection_id": "a289d9ea-e623-480f-81cc-99baa9d6733c"     },     {       "x": 428.5,       "y": 395.5,       "width": 69,       "height": 113,       "confidence": 0.875,       "class": "door",       "class_id": 5,       "detection_id": "34ebf0f8-68ba-464e-9d0d-908ac165f0ad"     },     {       "x": 251,       "y": 583.5,       "width": 60,       "height": 113,       "confidence": 0.859,       "class": "door",       "class_id": 5,       "detection_id": "f6e28cdc-e8e5-4f38-9336-1c8584e096eb"     },     {       "x": 595.5,       "y": 430.5,       "width": 49,       "height": 67,       "confidence": 0.807,       "class": "toilet",       "class_id": 11,       "detection_id": "1c7f0b5a-6ad9-44f1-a69f-5f950af2fb35"     },     {       "x": 360.5,       "y": 177.5,       "width": 79,       "height": 101,       "confidence": 0.801,       "class": "bed",       "class_id": 2,       "detection_id": "ca1a2bd2-c15e-480d-952f-dae5ecb239c5"     }   ] }

推理结果:

{   "predictions": [     {       "x": 523,       "y": 327.5,       "width": 28,       "height": 49,       "confidence": 0.93,       "class": "ctable",       "class_id": 4,       "detection_id": "146f4cb7-2a4b-40c6-b820-236a457da41e"     },     {       "x": 244.5,       "y": 579.5,       "width": 51,       "height": 55,       "confidence": 0.922,       "class": "bed",       "class_id": 2,       "detection_id": "e60e12b1-d1f4-4fc8-b2a1-a489d979910c"     },     {       "x": 279.5,       "y": 125.5,       "width": 63,       "height": 89,       "confidence": 0.913,       "class": "dtable",       "class_id": 6,       "detection_id": "d406b8ff-8f98-4d11-ab45-731ae1e4ae2f"     },     {       "x": 244.5,       "y": 343.5,       "width": 63,       "height": 89,       "confidence": 0.912,       "class": "dtable",       "class_id": 6,       "detection_id": "afaed4b6-1237-4b7e-98f0-0caa5b60bf75"     },     {       "x": 597.5,       "y": 453.5,       "width": 35,       "height": 87,       "confidence": 0.91,       "class": "bed",       "class_id": 2,       "detection_id": "bb0f5c3c-6473-4bcc-a9cb-0d073672862d"     },     {       "x": 437,       "y": 453.5,       "width": 36,       "height": 87,       "confidence": 0.908,       "class": "bed",       "class_id": 2,       "detection_id": "dfba14f1-fa9a-4510-a4b3-55f9b349a05c"     },     {       "x": 525,       "y": 203.5,       "width": 24,       "height": 85,       "confidence": 0.908,       "class": "bathtub",       "class_id": 1,       "detection_id": "c63c76ce-9abb-48fe-a17f-ba454494f0e4"     },     {       "x": 545,       "y": 453,       "width": 36,       "height": 86,       "confidence": 0.904,       "class": "bed",       "class_id": 2,       "detection_id": "7634a0e6-fa93-4cf9-8796-bedc729b4c62"     },     {       "x": 523,       "y": 68,       "width": 26,       "height": 88,       "confidence": 0.903,       "class": "bathtub",       "class_id": 1,       "detection_id": "dfd89ffd-1c6d-45f4-b04e-f744bda18002"     },     {       "x": 541.5,       "y": 595.5,       "width": 31,       "height": 51,       "confidence": 0.902,       "class": "ctable",       "class_id": 4,       "detection_id": "3eb3ca0c-a30e-4656-ba22-388aaa55d5f5"     },     {       "x": 491.5,       "y": 451.5,       "width": 35,       "height": 85,       "confidence": 0.902,       "class": "bed",       "class_id": 2,       "detection_id": "9bb9a427-6013-48ef-baf5-04cc2cfc3a01"     },     {       "x": 453.5,       "y": 594.5,       "width": 29,       "height": 49,       "confidence": 0.889,       "class": "ctable",       "class_id": 4,       "detection_id": "372eedfe-7bd1-430a-a2e6-e103b21879f1"     },     {       "x": 124.5,       "y": 119.5,       "width": 27,       "height": 51,       "confidence": 0.884,       "class": "ctable",       "class_id": 4,       "detection_id": "dddaa611-03d6-4ad4-8da8-8ac9205cb229"     },     {       "x": 288,       "y": 251,       "width": 38,       "height": 66,       "confidence": 0.882,       "class": "door",       "class_id": 5,       "detection_id": "27db1168-baab-4a76-8173-4d32421475a2"     },     {       "x": 459.5,       "y": 252,       "width": 37,       "height": 64,       "confidence": 0.874,       "class": "door",       "class_id": 5,       "detection_id": "d97d831b-d950-438c-8d7b-b9d3bcc1837d"     },     {       "x": 497.5,       "y": 595,       "width": 31,       "height": 50,       "confidence": 0.873,       "class": "ctable",       "class_id": 4,       "detection_id": "6afe1d28-f4c9-4e8d-bd78-3d1bd6b12b11"     },     {       "x": 396.5,       "y": 45.5,       "width": 27,       "height": 41,       "confidence": 0.87,       "class": "toilet",       "class_id": 11,       "detection_id": "087a58a2-72e9-41bd-93f7-25a6e0ee86dd"     },     {       "x": 189,       "y": 414,       "width": 44,       "height": 68,       "confidence": 0.869,       "class": "door",       "class_id": 5,       "detection_id": "db083400-1f73-4bb3-8e60-4cb9362b431e"     },     {       "x": 395.5,       "y": 239,       "width": 27,       "height": 40,       "confidence": 0.865,       "class": "toilet",       "class_id": 11,       "detection_id": "815d3e0c-3f27-4529-9510-ea7d4a7123b1"     },     {       "x": 318.5,       "y": 582.5,       "width": 31,       "height": 51,       "confidence": 0.865,       "class": "chair",       "class_id": 3,       "detection_id": "ea4d5ee0-53f6-4086-88d4-94b20790b5ed"     },     {       "x": 159.5,       "y": 115.5,       "width": 31,       "height": 113,       "confidence": 0.859,       "class": "sofa",       "class_id": 8,       "detection_id": "274ab05b-ae04-4123-a033-d1471e0f7aad"     },     {       "x": 119,       "y": 248.5,       "width": 40,       "height": 67,       "confidence": 0.855,       "class": "door",       "class_id": 5,       "detection_id": "561f9eee-60b0-4631-a7c5-956469510388"     },     {       "x": 357,       "y": 495.5,       "width": 46,       "height": 65,       "confidence": 0.852,       "class": "door",       "class_id": 5,       "detection_id": "43374764-b6d8-4d0c-a3a7-e72213dfae00"     },     {       "x": 593.5,       "y": 598,       "width": 35,       "height": 56,       "confidence": 0.85,       "class": "ssofa",       "class_id": 10,       "detection_id": "d427e30f-a0a2-408d-a605-be3c2ca9cc26"     },     {       "x": 19.5,       "y": 119,       "width": 39,       "height": 66,       "confidence": 0.845,       "class": "door",       "class_id": 5,       "detection_id": "e2b5b57c-7a3f-400c-8356-11f1dd769b7f"     },     {       "x": 80.5,       "y": 115.5,       "width": 33,       "height": 113,       "confidence": 0.835,       "class": "sofa",       "class_id": 8,       "detection_id": "9b792622-1d2b-4c5e-88f0-130c4f61532b"     },     {       "x": 349,       "y": 67,       "width": 24,       "height": 72,       "confidence": 0.828,       "class": "sink",       "class_id": 7,       "detection_id": "53822221-72fb-42cf-9af1-312aaea3ad44"     },     {       "x": 240,       "y": 28.5,       "width": 50,       "height": 35,       "confidence": 0.81,       "class": "sink",       "class_id": 7,       "detection_id": "cd54f2b5-5be7-4eb7-9940-4a05be133027"     },     {       "x": 408,       "y": 591,       "width": 36,       "height": 58,       "confidence": 0.809,       "class": "bed",       "class_id": 2,       "detection_id": "ea52b384-956c-4781-a631-6c7938493cb3"     }   ] }

赞(0)
未经允许不得转载:171主机测评 » 带标注的家具识别数据集,可识别床,椅子,餐桌, 门,水槽,沙发,马桶,浴缸等12种,识别率87%,支持yolo,coco json,pascal voc xml格式的模型训练
分享到: 更多 (0)

评论 抢沙发

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址