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

模型训练指标参数:

模型训练图:

数据集拆分

训练集

3190图像

验证集

84图像

测试集

67图像

预处理

自动定向:应用

调整大小:拉伸到640x640

增强

每个训练样本的输出数量: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"
    }
  ]
}

Logo

AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。

更多推荐