{"id":78128,"date":"2026-02-26T08:51:09","date_gmt":"2026-02-26T00:51:09","guid":{"rendered":"https:\/\/www.wsisp.com\/helps\/78128.html"},"modified":"2026-02-26T08:51:09","modified_gmt":"2026-02-26T00:51:09","slug":"%e3%80%90%e5%ae%8c%e6%95%b4%e6%ba%90%e7%a0%81%e6%95%b0%e6%8d%ae%e9%9b%86%e9%83%a8%e7%bd%b2%e6%95%99%e7%a8%8b%e3%80%91%e4%b9%a1%e6%9d%91%e5%b0%8f%e9%81%93%e5%9b%be%e5%83%8f%e5%88%86%e5%89%b2%e7%b3%bb","status":"publish","type":"post","link":"https:\/\/www.wsisp.com\/helps\/78128.html","title":{"rendered":"\u3010\u5b8c\u6574\u6e90\u7801+\u6570\u636e\u96c6+\u90e8\u7f72\u6559\u7a0b\u3011\u4e61\u6751\u5c0f\u9053\u56fe\u50cf\u5206\u5272\u7cfb\u7edf\u6e90\u7801\uff06\u6570\u636e\u96c6\u5206\u4eab [yolov8-seg-C2f-Faster-EMA\uff06yolov8-seg-HGNetV2\u7b4950+\u5168\u5957\u6539\u8fdb\u521b\u65b0\u70b9\u53d1\u520a_\u4e00\u952e\u8bad\u7ec3\u6559\u7a0b"},"content":{"rendered":"<h2>\u80cc\u666f\u610f\u4e49<\/h2>\n<p>\u968f\u7740\u5168\u7403\u57ce\u9547\u5316\u8fdb\u7a0b\u7684\u52a0\u901f&#xff0c;\u4e61\u6751\u5730\u533a\u7684\u751f\u6001\u73af\u5883\u548c\u4ea4\u901a\u57fa\u7840\u8bbe\u65bd\u9762\u4e34\u7740\u524d\u6240\u672a\u6709\u7684\u6311\u6218\u3002\u4e61\u6751\u5c0f\u9053\u4f5c\u4e3a\u8fde\u63a5\u6751\u5e84\u4e0e\u5916\u754c\u7684\u91cd\u8981\u901a\u9053&#xff0c;\u4e0d\u4ec5\u627f\u8f7d\u7740\u519c\u6c11\u7684\u65e5\u5e38\u51fa\u884c\u548c\u519c\u4ea7\u54c1\u8fd0\u8f93&#xff0c;\u4e5f\u5728\u4e61\u6751\u65c5\u6e38\u548c\u751f\u6001\u4fdd\u62a4\u4e2d\u53d1\u6325\u7740\u91cd\u8981\u4f5c\u7528\u3002\u7136\u800c&#xff0c;\u4f20\u7edf\u7684\u4e61\u6751\u5c0f\u9053\u5f80\u5f80\u7f3a\u4e4f\u6709\u6548\u7684\u7ba1\u7406\u548c\u7ef4\u62a4&#xff0c;\u5bfc\u81f4\u9053\u8def\u635f\u574f\u3001\u690d\u88ab\u8fc7\u5ea6\u751f\u957f\u7b49\u95ee\u9898&#xff0c;\u4e25\u91cd\u5f71\u54cd\u4e86\u4e61\u6751\u7684\u53ef\u6301\u7eed\u53d1\u5c55\u3002\u56e0\u6b64&#xff0c;\u5982\u4f55\u6709\u6548\u5730\u76d1\u6d4b\u548c\u7ba1\u7406\u4e61\u6751\u5c0f\u9053&#xff0c;\u6210\u4e3a\u4e86\u5f53\u524d\u7814\u7a76\u7684\u70ed\u70b9\u4e4b\u4e00\u3002<\/p>\n<p>\u8fd1\u5e74\u6765&#xff0c;\u8ba1\u7b97\u673a\u89c6\u89c9\u6280\u672f\u7684\u5feb\u901f\u53d1\u5c55\u4e3a\u4e61\u6751\u5c0f\u9053\u7684\u76d1\u6d4b\u63d0\u4f9b\u4e86\u65b0\u7684\u89e3\u51b3\u65b9\u6848\u3002\u5c24\u5176\u662f\u57fa\u4e8e\u6df1\u5ea6\u5b66\u4e60\u7684\u56fe\u50cf\u5206\u5272\u6280\u672f&#xff0c;\u80fd\u591f\u5728\u590d\u6742\u7684\u81ea\u7136\u73af\u5883\u4e2d\u5b9e\u73b0\u9ad8\u7cbe\u5ea6\u7684\u76ee\u6807\u68c0\u6d4b\u4e0e\u5206\u5272\u3002YOLO&#xff08;You Only Look Once&#xff09;\u7cfb\u5217\u6a21\u578b\u56e0\u5176\u9ad8\u6548\u7684\u5b9e\u65f6\u68c0\u6d4b\u80fd\u529b&#xff0c;\u5df2\u5e7f\u6cdb\u5e94\u7528\u4e8e\u5404\u79cd\u573a\u666f\u3002\u7136\u800c&#xff0c;YOLOv8\u4f5c\u4e3a\u8be5\u7cfb\u5217\u7684\u6700\u65b0\u7248\u672c&#xff0c;\u867d\u7136\u5728\u7cbe\u5ea6\u548c\u901f\u5ea6\u4e0a\u90fd\u6709\u4e86\u663e\u8457\u63d0\u5347&#xff0c;\u4f46\u5728\u7279\u5b9a\u7684\u4e61\u6751\u5c0f\u9053\u73af\u5883\u4e0b&#xff0c;\u4ecd\u7136\u9762\u4e34\u7740\u4e00\u4e9b\u6311\u6218&#xff0c;\u5982\u590d\u6742\u80cc\u666f\u3001\u5149\u7167\u53d8\u5316\u548c\u7269\u4f53\u906e\u6321\u7b49\u95ee\u9898\u3002\u56e0\u6b64&#xff0c;\u6539\u8fdbYOLOv8\u4ee5\u9002\u5e94\u4e61\u6751\u5c0f\u9053\u56fe\u50cf\u5206\u5272\u7684\u9700\u6c42&#xff0c;\u5177\u6709\u91cd\u8981\u7684\u7406\u8bba\u548c\u5b9e\u8df5\u610f\u4e49\u3002<\/p>\n<p>\u672c\u7814\u7a76\u57fa\u4e8e\u201coffroad-dataset-ii-instance\u201d\u6570\u636e\u96c6&#xff0c;\u6784\u5efa\u4e86\u4e00\u5957\u9488\u5bf9\u4e61\u6751\u5c0f\u9053\u7684\u56fe\u50cf\u5206\u5272\u7cfb\u7edf\u3002\u8be5\u6570\u636e\u96c6\u5305\u542b1400\u5f20\u56fe\u50cf&#xff0c;\u6db5\u76d614\u4e2a\u7c7b\u522b&#xff0c;\u5305\u62ec\u80cc\u666f\u3001\u5bc6\u96c6\u690d\u88ab\u3001\u8349\u5730\u3001\u9ad8\u690d\u88ab\u3001\u4e0d\u53ef\u901a\u884c\u7684\u4f4e\u690d\u88ab\u3001\u7269\u4f53\u3001\u969c\u788d\u7269\u3001\u5c0f\u9053\u3001\u6c34\u5751\u3001\u7c97\u7cd9\u5c0f\u5f84\u3001\u5929\u7a7a\u3001\u5e73\u6ed1\u5c0f\u5f84\u3001\u53ef\u901a\u884c\u7684\u8349\u5730\u548c\u690d\u88ab\u7b49\u3002\u8fd9\u4e9b\u7c7b\u522b\u7684\u591a\u6837\u6027\u4f7f\u5f97\u6a21\u578b\u5728\u8bad\u7ec3\u8fc7\u7a0b\u4e2d\u80fd\u591f\u5b66\u4e60\u5230\u4e30\u5bcc\u7684\u7279\u5f81&#xff0c;\u4ece\u800c\u63d0\u9ad8\u5206\u5272\u7684\u51c6\u786e\u6027\u548c\u9c81\u68d2\u6027\u3002<\/p>\n<p>\u901a\u8fc7\u5bf9YOLOv8\u6a21\u578b\u7684\u6539\u8fdb&#xff0c;\u6211\u4eec\u671f\u671b\u80fd\u591f\u5728\u4e61\u6751\u5c0f\u9053\u7684\u56fe\u50cf\u5206\u5272\u4efb\u52a1\u4e2d&#xff0c;\u63d0\u5347\u6a21\u578b\u5bf9\u4e0d\u540c\u7c7b\u522b\u7684\u8bc6\u522b\u80fd\u529b&#xff0c;\u5c24\u5176\u662f\u5728\u590d\u6742\u73af\u5883\u4e0b\u7684\u8868\u73b0\u3002\u7814\u7a76\u7684\u610f\u4e49\u4e0d\u4ec5\u5728\u4e8e\u63a8\u52a8\u8ba1\u7b97\u673a\u89c6\u89c9\u6280\u672f\u5728\u4e61\u6751\u5c0f\u9053\u76d1\u6d4b\u4e2d\u7684\u5e94\u7528&#xff0c;\u8fd8\u5728\u4e8e\u4e3a\u4e61\u6751\u53ef\u6301\u7eed\u53d1\u5c55\u63d0\u4f9b\u6570\u636e\u652f\u6301\u548c\u51b3\u7b56\u4f9d\u636e\u3002\u901a\u8fc7\u7cbe\u786e\u7684\u56fe\u50cf\u5206\u5272&#xff0c;\u7ba1\u7406\u8005\u53ef\u4ee5\u66f4\u597d\u5730\u8bc6\u522b\u548c\u8bc4\u4f30\u4e61\u6751\u5c0f\u9053\u7684\u72b6\u51b5&#xff0c;\u5236\u5b9a\u76f8\u5e94\u7684\u7ef4\u62a4\u548c\u7ba1\u7406\u7b56\u7565&#xff0c;\u4ece\u800c\u4fc3\u8fdb\u4e61\u6751\u7ecf\u6d4e\u7684\u53d1\u5c55\u548c\u751f\u6001\u73af\u5883\u7684\u4fdd\u62a4\u3002<\/p>\n<p>\u6b64\u5916&#xff0c;\u672c\u7814\u7a76\u8fd8\u5c06\u4e3a\u76f8\u5173\u9886\u57df\u7684\u7814\u7a76\u63d0\u4f9b\u53c2\u8003&#xff0c;\u63a8\u52a8\u4e61\u6751\u4ea4\u901a\u57fa\u7840\u8bbe\u65bd\u667a\u80fd\u5316\u7ba1\u7406\u7684\u53d1\u5c55\u3002\u968f\u7740\u6280\u672f\u7684\u4e0d\u65ad\u8fdb\u6b65&#xff0c;\u672a\u6765\u7684\u4e61\u6751\u5c0f\u9053\u76d1\u6d4b\u7cfb\u7edf\u6709\u671b\u5b9e\u73b0\u66f4\u9ad8\u7684\u81ea\u52a8\u5316\u6c34\u5e73&#xff0c;\u4e3a\u4e61\u6751\u632f\u5174\u6218\u7565\u7684\u5b9e\u65bd\u63d0\u4f9b\u5f3a\u6709\u529b\u7684\u6280\u672f\u652f\u6301\u3002\u56e0\u6b64&#xff0c;\u57fa\u4e8e\u6539\u8fdbYOLOv8\u7684\u4e61\u6751\u5c0f\u9053\u56fe\u50cf\u5206\u5272\u7cfb\u7edf\u7684\u7814\u7a76&#xff0c;\u4e0d\u4ec5\u5177\u6709\u91cd\u8981\u7684\u5b66\u672f\u4ef7\u503c&#xff0c;\u4e5f\u5c06\u4e3a\u5b9e\u9645\u5e94\u7528\u5e26\u6765\u6df1\u8fdc\u7684\u5f71\u54cd\u3002<\/p>\n<h2>\u56fe\u7247\u6548\u679c<\/h2>\n<p><img decoding=\"async\" src=\"https:\/\/www.wsisp.com\/helps\/wp-content\/uploads\/2026\/02\/20260226005101-699f98f5b59df.jpg\" alt=\"\u5728\u8fd9\u91cc\u63d2\u5165\u56fe\u7247\u63cf\u8ff0\" \/><br \/>\n<img decoding=\"async\" src=\"https:\/\/www.wsisp.com\/helps\/wp-content\/uploads\/2026\/02\/20260226005103-699f98f76d7da.jpg\" alt=\"\u5728\u8fd9\u91cc\u63d2\u5165\u56fe\u7247\u63cf\u8ff0\" \/><br \/>\n<img decoding=\"async\" src=\"https:\/\/www.wsisp.com\/helps\/wp-content\/uploads\/2026\/02\/20260226005105-699f98f9462fc.jpg\" alt=\"\u5728\u8fd9\u91cc\u63d2\u5165\u56fe\u7247\u63cf\u8ff0\" \/><\/p>\n<h2>\u6570\u636e\u96c6\u4fe1\u606f<\/h2>\n<p>\u5728\u672c\u7814\u7a76\u4e2d&#xff0c;\u6211\u4eec\u91c7\u7528\u4e86\u540d\u4e3a\u201coffroad-dataset-ii-instance\u201d\u7684\u6570\u636e\u96c6&#xff0c;\u4ee5\u652f\u6301\u5bf9\u4e61\u6751\u5c0f\u9053\u56fe\u50cf\u5206\u5272\u7cfb\u7edf\u7684\u8bad\u7ec3&#xff0c;\u65e8\u5728\u6539\u8fdbYOLOv8-seg\u6a21\u578b\u7684\u6027\u80fd\u3002\u8be5\u6570\u636e\u96c6\u4e13\u6ce8\u4e8e\u4e61\u6751\u73af\u5883\u4e2d\u7684\u5c0f\u9053\u8bc6\u522b\u4e0e\u5206\u5272&#xff0c;\u5177\u6709\u91cd\u8981\u7684\u5e94\u7528\u4ef7\u503c&#xff0c;\u5c24\u5176\u662f\u5728\u81ea\u52a8\u9a7e\u9a76\u3001\u73af\u5883\u76d1\u6d4b\u548c\u519c\u4e1a\u7ba1\u7406\u7b49\u9886\u57df\u3002\u901a\u8fc7\u5bf9\u4e61\u6751\u5c0f\u9053\u7684\u7cbe\u51c6\u5206\u5272&#xff0c;\u6211\u4eec\u80fd\u591f\u4e3a\u76f8\u5173\u9886\u57df\u63d0\u4f9b\u66f4\u4e3a\u53ef\u9760\u7684\u6570\u636e\u652f\u6301&#xff0c;\u4ece\u800c\u63d0\u5347\u51b3\u7b56\u7684\u79d1\u5b66\u6027\u4e0e\u6709\u6548\u6027\u3002<\/p>\n<p>\u201coffroad-dataset-ii-instance\u201d\u6570\u636e\u96c6\u7684\u8bbe\u8ba1\u7406\u5ff5\u662f\u4e3a\u4e86\u89e3\u51b3\u4f20\u7edf\u56fe\u50cf\u5206\u5272\u4efb\u52a1\u4e2d\u5b58\u5728\u7684\u6311\u6218&#xff0c;\u5c24\u5176\u662f\u5728\u590d\u6742\u7684\u81ea\u7136\u73af\u5883\u4e2d\u3002\u8be5\u6570\u636e\u96c6\u7684\u7c7b\u522b\u6570\u91cf\u4e3a1&#xff0c;\u5177\u4f53\u7c7b\u522b\u4e3a\u201cpath\u201d&#xff0c;\u8fd9\u610f\u5473\u7740\u6570\u636e\u96c6\u4e13\u6ce8\u4e8e\u4e61\u6751\u5c0f\u9053\u7684\u5b9e\u4f8b\u5206\u5272\u3002\u5c3d\u7ba1\u7c7b\u522b\u6570\u91cf\u76f8\u5bf9\u8f83\u5c11&#xff0c;\u4f46\u8fd9\u4e00\u5355\u4e00\u7c7b\u522b\u7684\u6df1\u5ea6\u548c\u591a\u6837\u6027\u4f7f\u5f97\u6570\u636e\u96c6\u5728\u8bad\u7ec3\u6a21\u578b\u65f6\u80fd\u591f\u805a\u7126\u4e8e\u7279\u5b9a\u7684\u4efb\u52a1&#xff0c;\u51cf\u5c11\u4e86\u7c7b\u522b\u95f4\u7684\u5e72\u6270&#xff0c;\u8fdb\u800c\u63d0\u9ad8\u4e86\u6a21\u578b\u7684\u5b66\u4e60\u6548\u7387\u548c\u5206\u5272\u7cbe\u5ea6\u3002<\/p>\n<p>\u6570\u636e\u96c6\u4e2d\u7684\u56fe\u50cf\u6765\u6e90\u4e8e\u4e0d\u540c\u7684\u4e61\u6751\u573a\u666f&#xff0c;\u6db5\u76d6\u4e86\u591a\u79cd\u5929\u6c14\u6761\u4ef6\u3001\u5149\u7167\u53d8\u5316\u4ee5\u53ca\u4e0d\u540c\u7684\u5730\u5f62\u7279\u5f81\u3002\u8fd9\u79cd\u591a\u6837\u6027\u786e\u4fdd\u4e86\u6a21\u578b\u5728\u8bad\u7ec3\u8fc7\u7a0b\u4e2d\u80fd\u591f\u63a5\u89e6\u5230\u5404\u79cd\u53ef\u80fd\u7684\u73af\u5883\u56e0\u7d20&#xff0c;\u4ece\u800c\u589e\u5f3a\u5176\u5728\u5b9e\u9645\u5e94\u7528\u4e2d\u7684\u9c81\u68d2\u6027\u3002\u4f8b\u5982&#xff0c;\u6570\u636e\u96c6\u4e2d\u53ef\u80fd\u5305\u542b\u9633\u5149\u660e\u5a9a\u7684\u65e5\u5b50\u3001\u9634\u96e8\u5929\u6c14\u4ee5\u53ca\u4e0d\u540c\u5b63\u8282\u7684\u573a\u666f&#xff0c;\u8fd9\u4e9b\u56e0\u7d20\u90fd\u4f1a\u5f71\u54cd\u5c0f\u9053\u7684\u5916\u89c2\u548c\u53ef\u89c1\u6027\u3002\u901a\u8fc7\u8fd9\u79cd\u65b9\u5f0f&#xff0c;\u6a21\u578b\u80fd\u591f\u5b66\u4e60\u5230\u66f4\u4e3a\u5168\u9762\u7684\u7279\u5f81&#xff0c;\u4ece\u800c\u5728\u5b9e\u9645\u5e94\u7528\u4e2d\u8868\u73b0\u51fa\u66f4\u5f3a\u7684\u9002\u5e94\u80fd\u529b\u3002<\/p>\n<p>\u5728\u6570\u636e\u96c6\u7684\u6784\u5efa\u8fc7\u7a0b\u4e2d&#xff0c;\u6807\u6ce8\u56e2\u961f\u5bf9\u6bcf\u4e00\u5e45\u56fe\u50cf\u8fdb\u884c\u4e86\u7ec6\u81f4\u7684\u6807\u6ce8&#xff0c;\u786e\u4fdd\u6bcf\u4e2a\u5c0f\u9053\u7684\u8fb9\u754c\u6e05\u6670\u53ef\u8fa8\u3002\u8fd9\u79cd\u9ad8\u8d28\u91cf\u7684\u6807\u6ce8\u4e0d\u4ec5\u63d0\u9ad8\u4e86\u6570\u636e\u96c6\u7684\u53ef\u9760\u6027&#xff0c;\u4e5f\u4e3a\u540e\u7eed\u7684\u6a21\u578b\u8bad\u7ec3\u63d0\u4f9b\u4e86\u575a\u5b9e\u7684\u57fa\u7840\u3002\u6807\u6ce8\u7684\u51c6\u786e\u6027\u76f4\u63a5\u5f71\u54cd\u5230\u6a21\u578b\u7684\u8bad\u7ec3\u6548\u679c&#xff0c;\u56e0\u6b64\u5728\u6570\u636e\u96c6\u7684\u5236\u4f5c\u8fc7\u7a0b\u4e2d&#xff0c;\u56e2\u961f\u91c7\u7528\u4e86\u4e25\u683c\u7684\u8d28\u91cf\u63a7\u5236\u6807\u51c6&#xff0c;\u4ee5\u786e\u4fdd\u6bcf\u4e2a\u6807\u6ce8\u7684\u51c6\u786e\u6027\u548c\u4e00\u81f4\u6027\u3002<\/p>\n<p>\u4e3a\u4e86\u8fdb\u4e00\u6b65\u63d0\u5347\u6a21\u578b\u7684\u8bad\u7ec3\u6548\u679c&#xff0c;\u6211\u4eec\u5728\u6570\u636e\u9884\u5904\u7406\u9636\u6bb5\u5bf9\u56fe\u50cf\u8fdb\u884c\u4e86\u591a\u79cd\u589e\u5f3a\u64cd\u4f5c&#xff0c;\u5305\u62ec\u65cb\u8f6c\u3001\u7f29\u653e\u3001\u7ffb\u8f6c\u548c\u989c\u8272\u8c03\u6574\u7b49\u3002\u8fd9\u4e9b\u64cd\u4f5c\u4e0d\u4ec5\u589e\u52a0\u4e86\u6570\u636e\u96c6\u7684\u591a\u6837\u6027&#xff0c;\u8fd8\u5e2e\u52a9\u6a21\u578b\u66f4\u597d\u5730\u5b66\u4e60\u5230\u5c0f\u9053\u7684\u7279\u5f81&#xff0c;\u4f7f\u5176\u5728\u9762\u5bf9\u4e0d\u540c\u7684\u8f93\u5165\u65f6\u80fd\u591f\u4fdd\u6301\u8f83\u9ad8\u7684\u5206\u5272\u7cbe\u5ea6\u3002\u6b64\u5916&#xff0c;\u6570\u636e\u96c6\u8fd8\u63d0\u4f9b\u4e86\u591a\u79cd\u5206\u8fa8\u7387\u7684\u56fe\u50cf&#xff0c;\u4ee5\u9002\u5e94\u4e0d\u540c\u7684\u8bad\u7ec3\u9700\u6c42\u548c\u8ba1\u7b97\u8d44\u6e90\u3002<\/p>\n<p>\u603b\u4e4b&#xff0c;\u201coffroad-dataset-ii-instance\u201d\u6570\u636e\u96c6\u4e3a\u4e61\u6751\u5c0f\u9053\u56fe\u50cf\u5206\u5272\u7cfb\u7edf\u7684\u8bad\u7ec3\u63d0\u4f9b\u4e86\u4e30\u5bcc\u800c\u9ad8\u8d28\u91cf\u7684\u6570\u636e\u652f\u6301\u3002\u901a\u8fc7\u5bf9\u8be5\u6570\u636e\u96c6\u7684\u6df1\u5165\u5206\u6790\u4e0e\u5e94\u7528&#xff0c;\u6211\u4eec\u671f\u5f85\u80fd\u591f\u663e\u8457\u63d0\u5347YOLOv8-seg\u6a21\u578b\u5728\u4e61\u6751\u5c0f\u9053\u5206\u5272\u4efb\u52a1\u4e2d\u7684\u8868\u73b0&#xff0c;\u4e3a\u76f8\u5173\u9886\u57df\u7684\u7814\u7a76\u4e0e\u5e94\u7528\u63d0\u4f9b\u6709\u529b\u7684\u6280\u672f\u652f\u6301\u3002\u968f\u7740\u6a21\u578b\u7684\u4e0d\u65ad\u4f18\u5316\u4e0e\u8fed\u4ee3&#xff0c;\u6211\u4eec\u76f8\u4fe1\u8fd9\u4e00\u6570\u636e\u96c6\u5c06\u4e3a\u672a\u6765\u7684\u7814\u7a76\u5960\u5b9a\u575a\u5b9e\u7684\u57fa\u7840&#xff0c;\u63a8\u52a8\u4e61\u6751\u73af\u5883\u667a\u80fd\u5316\u7ba1\u7406\u7684\u53d1\u5c55\u3002<br \/>\n<img decoding=\"async\" src=\"https:\/\/www.wsisp.com\/helps\/wp-content\/uploads\/2026\/02\/20260226005107-699f98fb2ae34.jpg\" alt=\"\u5728\u8fd9\u91cc\u63d2\u5165\u56fe\u7247\u63cf\u8ff0\" \/><br \/>\n<img decoding=\"async\" src=\"https:\/\/www.wsisp.com\/helps\/wp-content\/uploads\/2026\/02\/20260226005107-699f98fb56ddb.jpg\" alt=\"\u5728\u8fd9\u91cc\u63d2\u5165\u56fe\u7247\u63cf\u8ff0\" \/><br \/>\n<img decoding=\"async\" src=\"https:\/\/www.wsisp.com\/helps\/wp-content\/uploads\/2026\/02\/20260226005107-699f98fb81fd7.jpg\" alt=\"\u5728\u8fd9\u91cc\u63d2\u5165\u56fe\u7247\u63cf\u8ff0\" \/><br \/>\n<img decoding=\"async\" src=\"https:\/\/www.wsisp.com\/helps\/wp-content\/uploads\/2026\/02\/20260226005107-699f98fbae020.jpg\" alt=\"\u5728\u8fd9\u91cc\u63d2\u5165\u56fe\u7247\u63cf\u8ff0\" \/><br \/>\n<img decoding=\"async\" src=\"https:\/\/www.wsisp.com\/helps\/wp-content\/uploads\/2026\/02\/20260226005107-699f98fbda201.jpg\" alt=\"\u5728\u8fd9\u91cc\u63d2\u5165\u56fe\u7247\u63cf\u8ff0\" \/><\/p>\n<h2>\u6838\u5fc3\u4ee3\u7801<\/h2>\n<p>&#096;&#096;&#096;\u4ee5\u4e0b\u662f\u5bf9\u4ee3\u7801\u7684\u6838\u5fc3\u90e8\u5206\u8fdb\u884c\u7684\u5206\u6790\u548c\u8be6\u7ec6\u6ce8\u91ca\u3002\u4ee3\u7801\u4e3b\u8981\u662f\u4e00\u4e2a\u7528\u4e8e\u52a0\u8f7d\u548c\u5904\u7406\u56fe\u50cf\u6570\u636e\u7684\u57fa\u7840\u6570\u636e\u96c6\u7c7b BaseDataset&#xff0c;\u5b83\u662f PyTorch \u7684 Dataset \u7c7b\u7684\u5b50\u7c7b\u3002<\/p>\n<p><span class=\"token keyword\">import<\/span> glob<br \/>\n<span class=\"token keyword\">import<\/span> math<br \/>\n<span class=\"token keyword\">import<\/span> os<br \/>\n<span class=\"token keyword\">import<\/span> random<br \/>\n<span class=\"token keyword\">from<\/span> copy <span class=\"token keyword\">import<\/span> deepcopy<br \/>\n<span class=\"token keyword\">from<\/span> multiprocessing<span class=\"token punctuation\">.<\/span>pool <span class=\"token keyword\">import<\/span> ThreadPool<br \/>\n<span class=\"token keyword\">from<\/span> pathlib <span class=\"token keyword\">import<\/span> Path<br \/>\n<span class=\"token keyword\">from<\/span> typing <span class=\"token keyword\">import<\/span> Optional<\/p>\n<p><span class=\"token keyword\">import<\/span> cv2<br \/>\n<span class=\"token keyword\">import<\/span> numpy <span class=\"token keyword\">as<\/span> np<br \/>\n<span class=\"token keyword\">import<\/span> psutil<br \/>\n<span class=\"token keyword\">from<\/span> torch<span class=\"token punctuation\">.<\/span>utils<span class=\"token punctuation\">.<\/span>data <span class=\"token keyword\">import<\/span> Dataset<\/p>\n<p><span class=\"token keyword\">from<\/span> ultralytics<span class=\"token punctuation\">.<\/span>utils <span class=\"token keyword\">import<\/span> DEFAULT_CFG<span class=\"token punctuation\">,<\/span> LOCAL_RANK<span class=\"token punctuation\">,<\/span> LOGGER<span class=\"token punctuation\">,<\/span> NUM_THREADS<span class=\"token punctuation\">,<\/span> TQDM<br \/>\n<span class=\"token keyword\">from<\/span> <span class=\"token punctuation\">.<\/span>utils <span class=\"token keyword\">import<\/span> HELP_URL<span class=\"token punctuation\">,<\/span> IMG_FORMATS<\/p>\n<p><span class=\"token keyword\">class<\/span> <span class=\"token class-name\">BaseDataset<\/span><span class=\"token punctuation\">(<\/span>Dataset<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n    <span class=\"token triple-quoted-string string\">&#034;&#034;&#034;<br \/>\n    \u57fa\u7840\u6570\u636e\u96c6\u7c7b&#xff0c;\u7528\u4e8e\u52a0\u8f7d\u548c\u5904\u7406\u56fe\u50cf\u6570\u636e\u3002<\/p>\n<p>    \u53c2\u6570:<br \/>\n        img_path (str): \u5305\u542b\u56fe\u50cf\u7684\u6587\u4ef6\u5939\u8def\u5f84\u3002<br \/>\n        imgsz (int, optional): \u56fe\u50cf\u5927\u5c0f&#xff0c;\u9ed8\u8ba4\u4e3a640\u3002<br \/>\n        cache (bool, optional): \u5728\u8bad\u7ec3\u671f\u95f4\u5c06\u56fe\u50cf\u7f13\u5b58\u5230RAM\u6216\u78c1\u76d8&#xff0c;\u9ed8\u8ba4\u4e3aFalse\u3002<br \/>\n        augment (bool, optional): \u5982\u679c\u4e3aTrue&#xff0c;\u5219\u5e94\u7528\u6570\u636e\u589e\u5f3a&#xff0c;\u9ed8\u8ba4\u4e3aTrue\u3002<br \/>\n        hyp (dict, optional): \u5e94\u7528\u6570\u636e\u589e\u5f3a\u7684\u8d85\u53c2\u6570&#xff0c;\u9ed8\u8ba4\u4e3aNone\u3002<br \/>\n        prefix (str, optional): \u65e5\u5fd7\u6d88\u606f\u4e2d\u6253\u5370\u7684\u524d\u7f00&#xff0c;\u9ed8\u8ba4\u4e3a&#039;&#039;\u3002<br \/>\n        rect (bool, optional): \u5982\u679c\u4e3aTrue&#xff0c;\u5219\u4f7f\u7528\u77e9\u5f62\u8bad\u7ec3&#xff0c;\u9ed8\u8ba4\u4e3aFalse\u3002<br \/>\n        batch_size (int, optional): \u6279\u6b21\u5927\u5c0f&#xff0c;\u9ed8\u8ba4\u4e3aNone\u3002<br \/>\n        stride (int, optional): \u6b65\u5e45&#xff0c;\u9ed8\u8ba4\u4e3a32\u3002<br \/>\n        pad (float, optional): \u586b\u5145&#xff0c;\u9ed8\u8ba4\u4e3a0.0\u3002<br \/>\n        single_cls (bool, optional): \u5982\u679c\u4e3aTrue&#xff0c;\u5219\u4f7f\u7528\u5355\u7c7b\u8bad\u7ec3&#xff0c;\u9ed8\u8ba4\u4e3aFalse\u3002<br \/>\n        classes (list): \u5305\u542b\u7684\u7c7b\u7684\u5217\u8868&#xff0c;\u9ed8\u8ba4\u4e3aNone\u3002<br \/>\n        fraction (float): \u4f7f\u7528\u7684\u6570\u636e\u96c6\u7684\u6bd4\u4f8b&#xff0c;\u9ed8\u8ba4\u4e3a1.0&#xff08;\u4f7f\u7528\u6240\u6709\u6570\u636e&#xff09;\u3002<\/p>\n<p>    \u5c5e\u6027:<br \/>\n        im_files (list): \u56fe\u50cf\u6587\u4ef6\u8def\u5f84\u5217\u8868\u3002<br \/>\n        labels (list): \u6807\u7b7e\u6570\u636e\u5b57\u5178\u5217\u8868\u3002<br \/>\n        ni (int): \u6570\u636e\u96c6\u4e2d\u56fe\u50cf\u7684\u6570\u91cf\u3002<br \/>\n        ims (list): \u52a0\u8f7d\u7684\u56fe\u50cf\u5217\u8868\u3002<br \/>\n        npy_files (list): numpy\u6587\u4ef6\u8def\u5f84\u5217\u8868\u3002<br \/>\n        transforms (callable): \u56fe\u50cf\u8f6c\u6362\u51fd\u6570\u3002<br \/>\n    &#034;&#034;&#034;<\/span><\/p>\n<p>    <span class=\"token keyword\">def<\/span> <span class=\"token function\">__init__<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> img_path<span class=\"token punctuation\">,<\/span> imgsz<span class=\"token operator\">&#061;<\/span><span class=\"token number\">640<\/span><span class=\"token punctuation\">,<\/span> cache<span class=\"token operator\">&#061;<\/span><span class=\"token boolean\">False<\/span><span class=\"token punctuation\">,<\/span> augment<span class=\"token operator\">&#061;<\/span><span class=\"token boolean\">True<\/span><span class=\"token punctuation\">,<\/span> hyp<span class=\"token operator\">&#061;<\/span>DEFAULT_CFG<span class=\"token punctuation\">,<\/span> prefix<span class=\"token operator\">&#061;<\/span><span class=\"token string\">&#039;&#039;<\/span><span class=\"token punctuation\">,<\/span> rect<span class=\"token operator\">&#061;<\/span><span class=\"token boolean\">False<\/span><span class=\"token punctuation\">,<\/span><br \/>\n                 batch_size<span class=\"token operator\">&#061;<\/span><span class=\"token number\">16<\/span><span class=\"token punctuation\">,<\/span> stride<span class=\"token operator\">&#061;<\/span><span class=\"token number\">32<\/span><span class=\"token punctuation\">,<\/span> pad<span class=\"token operator\">&#061;<\/span><span class=\"token number\">0.5<\/span><span class=\"token punctuation\">,<\/span> single_cls<span class=\"token operator\">&#061;<\/span><span class=\"token boolean\">False<\/span><span class=\"token punctuation\">,<\/span> classes<span class=\"token operator\">&#061;<\/span><span class=\"token boolean\">None<\/span><span class=\"token punctuation\">,<\/span> fraction<span class=\"token operator\">&#061;<\/span><span class=\"token number\">1.0<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n        <span class=\"token triple-quoted-string string\">&#034;&#034;&#034;\u4f7f\u7528\u7ed9\u5b9a\u7684\u914d\u7f6e\u548c\u9009\u9879\u521d\u59cb\u5316BaseDataset\u3002&#034;&#034;&#034;<\/span><br \/>\n        <span class=\"token builtin\">super<\/span><span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">.<\/span>__init__<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span>img_path <span class=\"token operator\">&#061;<\/span> img_path  <span class=\"token comment\"># \u56fe\u50cf\u8def\u5f84<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span>imgsz <span class=\"token operator\">&#061;<\/span> imgsz  <span class=\"token comment\"># \u56fe\u50cf\u5927\u5c0f<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span>augment <span class=\"token operator\">&#061;<\/span> augment  <span class=\"token comment\"># \u662f\u5426\u5e94\u7528\u6570\u636e\u589e\u5f3a<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span>single_cls <span class=\"token operator\">&#061;<\/span> single_cls  <span class=\"token comment\"># \u662f\u5426\u4f7f\u7528\u5355\u7c7b\u8bad\u7ec3<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span>prefix <span class=\"token operator\">&#061;<\/span> prefix  <span class=\"token comment\"># \u65e5\u5fd7\u524d\u7f00<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span>fraction <span class=\"token operator\">&#061;<\/span> fraction  <span class=\"token comment\"># \u4f7f\u7528\u7684\u6570\u636e\u96c6\u6bd4\u4f8b<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span>im_files <span class=\"token operator\">&#061;<\/span> self<span class=\"token punctuation\">.<\/span>get_img_files<span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">.<\/span>img_path<span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u83b7\u53d6\u56fe\u50cf\u6587\u4ef6<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span>labels <span class=\"token operator\">&#061;<\/span> self<span class=\"token punctuation\">.<\/span>get_labels<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u83b7\u53d6\u6807\u7b7e<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span>update_labels<span class=\"token punctuation\">(<\/span>include_class<span class=\"token operator\">&#061;<\/span>classes<span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u66f4\u65b0\u6807\u7b7e\u4ee5\u4ec5\u5305\u542b\u6307\u5b9a\u7c7b<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span>ni <span class=\"token operator\">&#061;<\/span> <span class=\"token builtin\">len<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">.<\/span>labels<span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u6570\u636e\u96c6\u4e2d\u56fe\u50cf\u7684\u6570\u91cf<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span>rect <span class=\"token operator\">&#061;<\/span> rect  <span class=\"token comment\"># \u662f\u5426\u4f7f\u7528\u77e9\u5f62\u8bad\u7ec3<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span>batch_size <span class=\"token operator\">&#061;<\/span> batch_size  <span class=\"token comment\"># \u6279\u6b21\u5927\u5c0f<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span>stride <span class=\"token operator\">&#061;<\/span> stride  <span class=\"token comment\"># \u6b65\u5e45<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span>pad <span class=\"token operator\">&#061;<\/span> pad  <span class=\"token comment\"># \u586b\u5145<\/span><br \/>\n        <span class=\"token keyword\">if<\/span> self<span class=\"token punctuation\">.<\/span>rect<span class=\"token punctuation\">:<\/span><br \/>\n            <span class=\"token keyword\">assert<\/span> self<span class=\"token punctuation\">.<\/span>batch_size <span class=\"token keyword\">is<\/span> <span class=\"token keyword\">not<\/span> <span class=\"token boolean\">None<\/span><br \/>\n            self<span class=\"token punctuation\">.<\/span>set_rectangle<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u8bbe\u7f6e\u77e9\u5f62\u5f62\u72b6<\/span><\/p>\n<p>        <span class=\"token comment\"># \u7528\u4e8e\u7f13\u5b58\u56fe\u50cf\u7684\u7f13\u51b2\u533a<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span><span class=\"token builtin\">buffer<\/span> <span class=\"token operator\">&#061;<\/span> <span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">]<\/span>  <span class=\"token comment\"># \u7f13\u51b2\u533a\u5927\u5c0f &#061; \u6279\u6b21\u5927\u5c0f<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span>max_buffer_length <span class=\"token operator\">&#061;<\/span> <span class=\"token builtin\">min<\/span><span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">.<\/span>ni<span class=\"token punctuation\">,<\/span> self<span class=\"token punctuation\">.<\/span>batch_size <span class=\"token operator\">*<\/span> <span class=\"token number\">8<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token number\">1000<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">)<\/span> <span class=\"token keyword\">if<\/span> self<span class=\"token punctuation\">.<\/span>augment <span class=\"token keyword\">else<\/span> <span class=\"token number\">0<\/span><\/p>\n<p>        <span class=\"token comment\"># \u7f13\u5b58\u56fe\u50cf<\/span><br \/>\n        <span class=\"token keyword\">if<\/span> cache <span class=\"token operator\">&#061;&#061;<\/span> <span class=\"token string\">&#039;ram&#039;<\/span> <span class=\"token keyword\">and<\/span> <span class=\"token keyword\">not<\/span> self<span class=\"token punctuation\">.<\/span>check_cache_ram<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n            cache <span class=\"token operator\">&#061;<\/span> <span class=\"token boolean\">False<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span>ims<span class=\"token punctuation\">,<\/span> self<span class=\"token punctuation\">.<\/span>im_hw0<span class=\"token punctuation\">,<\/span> self<span class=\"token punctuation\">.<\/span>im_hw <span class=\"token operator\">&#061;<\/span> <span class=\"token punctuation\">[<\/span><span class=\"token boolean\">None<\/span><span class=\"token punctuation\">]<\/span> <span class=\"token operator\">*<\/span> self<span class=\"token punctuation\">.<\/span>ni<span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">[<\/span><span class=\"token boolean\">None<\/span><span class=\"token punctuation\">]<\/span> <span class=\"token operator\">*<\/span> self<span class=\"token punctuation\">.<\/span>ni<span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">[<\/span><span class=\"token boolean\">None<\/span><span class=\"token punctuation\">]<\/span> <span class=\"token operator\">*<\/span> self<span class=\"token punctuation\">.<\/span>ni<br \/>\n        self<span class=\"token punctuation\">.<\/span>npy_files <span class=\"token operator\">&#061;<\/span> <span class=\"token punctuation\">[<\/span>Path<span class=\"token punctuation\">(<\/span>f<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">.<\/span>with_suffix<span class=\"token punctuation\">(<\/span><span class=\"token string\">&#039;.npy&#039;<\/span><span class=\"token punctuation\">)<\/span> <span class=\"token keyword\">for<\/span> f <span class=\"token keyword\">in<\/span> self<span class=\"token punctuation\">.<\/span>im_files<span class=\"token punctuation\">]<\/span>  <span class=\"token comment\"># \u5c06\u56fe\u50cf\u6587\u4ef6\u8def\u5f84\u8f6c\u6362\u4e3anpy\u6587\u4ef6\u8def\u5f84<\/span><br \/>\n        <span class=\"token keyword\">if<\/span> cache<span class=\"token punctuation\">:<\/span><br \/>\n            self<span class=\"token punctuation\">.<\/span>cache_images<span class=\"token punctuation\">(<\/span>cache<span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u7f13\u5b58\u56fe\u50cf<\/span><\/p>\n<p>        <span class=\"token comment\"># \u56fe\u50cf\u8f6c\u6362<\/span><br \/>\n        self<span class=\"token punctuation\">.<\/span>transforms <span class=\"token operator\">&#061;<\/span> self<span class=\"token punctuation\">.<\/span>build_transforms<span class=\"token punctuation\">(<\/span>hyp<span class=\"token operator\">&#061;<\/span>hyp<span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u6784\u5efa\u56fe\u50cf\u8f6c\u6362<\/span><\/p>\n<p>    <span class=\"token keyword\">def<\/span> <span class=\"token function\">get_img_files<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> img_path<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n        <span class=\"token triple-quoted-string string\">&#034;&#034;&#034;\u8bfb\u53d6\u56fe\u50cf\u6587\u4ef6\u3002&#034;&#034;&#034;<\/span><br \/>\n        <span class=\"token keyword\">try<\/span><span class=\"token punctuation\">:<\/span><br \/>\n            f <span class=\"token operator\">&#061;<\/span> <span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">]<\/span>  <span class=\"token comment\"># \u56fe\u50cf\u6587\u4ef6\u5217\u8868<\/span><br \/>\n            <span class=\"token keyword\">for<\/span> p <span class=\"token keyword\">in<\/span> img_path <span class=\"token keyword\">if<\/span> <span class=\"token builtin\">isinstance<\/span><span class=\"token punctuation\">(<\/span>img_path<span class=\"token punctuation\">,<\/span> <span class=\"token builtin\">list<\/span><span class=\"token punctuation\">)<\/span> <span class=\"token keyword\">else<\/span> <span class=\"token punctuation\">[<\/span>img_path<span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">:<\/span><br \/>\n                p <span class=\"token operator\">&#061;<\/span> Path<span class=\"token punctuation\">(<\/span>p<span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u5904\u7406\u8def\u5f84<\/span><br \/>\n                <span class=\"token keyword\">if<\/span> p<span class=\"token punctuation\">.<\/span>is_dir<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>  <span class=\"token comment\"># \u5982\u679c\u662f\u76ee\u5f55<\/span><br \/>\n                    f <span class=\"token operator\">&#043;&#061;<\/span> glob<span class=\"token punctuation\">.<\/span>glob<span class=\"token punctuation\">(<\/span><span class=\"token builtin\">str<\/span><span class=\"token punctuation\">(<\/span>p <span class=\"token operator\">\/<\/span> <span class=\"token string\">&#039;**&#039;<\/span> <span class=\"token operator\">\/<\/span> <span class=\"token string\">&#039;*.*&#039;<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">,<\/span> recursive<span class=\"token operator\">&#061;<\/span><span class=\"token boolean\">True<\/span><span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u83b7\u53d6\u76ee\u5f55\u4e0b\u6240\u6709\u56fe\u50cf\u6587\u4ef6<\/span><br \/>\n                <span class=\"token keyword\">elif<\/span> p<span class=\"token punctuation\">.<\/span>is_file<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>  <span class=\"token comment\"># \u5982\u679c\u662f\u6587\u4ef6<\/span><br \/>\n                    <span class=\"token keyword\">with<\/span> <span class=\"token builtin\">open<\/span><span class=\"token punctuation\">(<\/span>p<span class=\"token punctuation\">)<\/span> <span class=\"token keyword\">as<\/span> t<span class=\"token punctuation\">:<\/span><br \/>\n                        t <span class=\"token operator\">&#061;<\/span> t<span class=\"token punctuation\">.<\/span>read<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">.<\/span>strip<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">.<\/span>splitlines<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u8bfb\u53d6\u6587\u4ef6\u5185\u5bb9<\/span><br \/>\n                        parent <span class=\"token operator\">&#061;<\/span> <span class=\"token builtin\">str<\/span><span class=\"token punctuation\">(<\/span>p<span class=\"token punctuation\">.<\/span>parent<span class=\"token punctuation\">)<\/span> <span class=\"token operator\">&#043;<\/span> os<span class=\"token punctuation\">.<\/span>sep<br \/>\n                        f <span class=\"token operator\">&#043;&#061;<\/span> <span class=\"token punctuation\">[<\/span>x<span class=\"token punctuation\">.<\/span>replace<span class=\"token punctuation\">(<\/span><span class=\"token string\">&#039;.\/&#039;<\/span><span class=\"token punctuation\">,<\/span> parent<span class=\"token punctuation\">)<\/span> <span class=\"token keyword\">if<\/span> x<span class=\"token punctuation\">.<\/span>startswith<span class=\"token punctuation\">(<\/span><span class=\"token string\">&#039;.\/&#039;<\/span><span class=\"token punctuation\">)<\/span> <span class=\"token keyword\">else<\/span> x <span class=\"token keyword\">for<\/span> x <span class=\"token keyword\">in<\/span> t<span class=\"token punctuation\">]<\/span>  <span class=\"token comment\"># \u8f6c\u6362\u4e3a\u5168\u5c40\u8def\u5f84<\/span><br \/>\n                <span class=\"token keyword\">else<\/span><span class=\"token punctuation\">:<\/span><br \/>\n                    <span class=\"token keyword\">raise<\/span> FileNotFoundError<span class=\"token punctuation\">(<\/span><span class=\"token string-interpolation\"><span class=\"token string\">f&#039;<\/span><span class=\"token interpolation\"><span class=\"token punctuation\">{<\/span>self<span class=\"token punctuation\">.<\/span>prefix<span class=\"token punctuation\">}<\/span><\/span><span class=\"token interpolation\"><span class=\"token punctuation\">{<\/span>p<span class=\"token punctuation\">}<\/span><\/span><span class=\"token string\"> does not exist&#039;<\/span><\/span><span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u6587\u4ef6\u4e0d\u5b58\u5728<\/span><br \/>\n            im_files <span class=\"token operator\">&#061;<\/span> <span class=\"token builtin\">sorted<\/span><span class=\"token punctuation\">(<\/span>x<span class=\"token punctuation\">.<\/span>replace<span class=\"token punctuation\">(<\/span><span class=\"token string\">&#039;\/&#039;<\/span><span class=\"token punctuation\">,<\/span> os<span class=\"token punctuation\">.<\/span>sep<span class=\"token punctuation\">)<\/span> <span class=\"token keyword\">for<\/span> x <span class=\"token keyword\">in<\/span> f <span class=\"token keyword\">if<\/span> x<span class=\"token punctuation\">.<\/span>split<span class=\"token punctuation\">(<\/span><span class=\"token string\">&#039;.&#039;<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">[<\/span><span class=\"token operator\">&#8211;<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">.<\/span>lower<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span> <span class=\"token keyword\">in<\/span> IMG_FORMATS<span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u8fc7\u6ee4\u51fa\u56fe\u50cf\u6587\u4ef6<\/span><br \/>\n            <span class=\"token keyword\">assert<\/span> im_files<span class=\"token punctuation\">,<\/span> <span class=\"token string-interpolation\"><span class=\"token string\">f&#039;<\/span><span class=\"token interpolation\"><span class=\"token punctuation\">{<\/span>self<span class=\"token punctuation\">.<\/span>prefix<span class=\"token punctuation\">}<\/span><\/span><span class=\"token string\">No images found in <\/span><span class=\"token interpolation\"><span class=\"token punctuation\">{<\/span>img_path<span class=\"token punctuation\">}<\/span><\/span><span class=\"token string\">&#039;<\/span><\/span>  <span class=\"token comment\"># \u786e\u4fdd\u627e\u5230\u56fe\u50cf\u6587\u4ef6<\/span><br \/>\n        <span class=\"token keyword\">except<\/span> Exception <span class=\"token keyword\">as<\/span> e<span class=\"token punctuation\">:<\/span><br \/>\n            <span class=\"token keyword\">raise<\/span> FileNotFoundError<span class=\"token punctuation\">(<\/span><span class=\"token string-interpolation\"><span class=\"token string\">f&#039;<\/span><span class=\"token interpolation\"><span class=\"token punctuation\">{<\/span>self<span class=\"token punctuation\">.<\/span>prefix<span class=\"token punctuation\">}<\/span><\/span><span class=\"token string\">Error loading data from <\/span><span class=\"token interpolation\"><span class=\"token punctuation\">{<\/span>img_path<span class=\"token punctuation\">}<\/span><\/span><span class=\"token string\">\\\\n<\/span><span class=\"token interpolation\"><span class=\"token punctuation\">{<\/span>HELP_URL<span class=\"token punctuation\">}<\/span><\/span><span class=\"token string\">&#039;<\/span><\/span><span class=\"token punctuation\">)<\/span> <span class=\"token keyword\">from<\/span> e<br \/>\n        <span class=\"token keyword\">if<\/span> self<span class=\"token punctuation\">.<\/span>fraction <span class=\"token operator\">&lt;<\/span> <span class=\"token number\">1<\/span><span class=\"token punctuation\">:<\/span><br \/>\n            im_files <span class=\"token operator\">&#061;<\/span> im_files<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token builtin\">round<\/span><span class=\"token punctuation\">(<\/span><span class=\"token builtin\">len<\/span><span class=\"token punctuation\">(<\/span>im_files<span class=\"token punctuation\">)<\/span> <span class=\"token operator\">*<\/span> self<span class=\"token punctuation\">.<\/span>fraction<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">]<\/span>  <span class=\"token comment\"># \u6839\u636e\u6bd4\u4f8b\u7b5b\u9009\u56fe\u50cf\u6587\u4ef6<\/span><br \/>\n        <span class=\"token keyword\">return<\/span> im_files<\/p>\n<p>    <span class=\"token keyword\">def<\/span> <span class=\"token function\">load_image<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> i<span class=\"token punctuation\">,<\/span> rect_mode<span class=\"token operator\">&#061;<\/span><span class=\"token boolean\">True<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n        <span class=\"token triple-quoted-string string\">&#034;&#034;&#034;\u4ece\u6570\u636e\u96c6\u4e2d\u52a0\u8f7d\u4e00\u5f20\u56fe\u50cf&#xff0c;\u8fd4\u56de (im, resized hw)\u3002&#034;&#034;&#034;<\/span><br \/>\n        im<span class=\"token punctuation\">,<\/span> f<span class=\"token punctuation\">,<\/span> fn <span class=\"token operator\">&#061;<\/span> self<span class=\"token punctuation\">.<\/span>ims<span class=\"token punctuation\">[<\/span>i<span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span> self<span class=\"token punctuation\">.<\/span>im_files<span class=\"token punctuation\">[<\/span>i<span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span> self<span class=\"token punctuation\">.<\/span>npy_files<span class=\"token punctuation\">[<\/span>i<span class=\"token punctuation\">]<\/span>  <span class=\"token comment\"># \u83b7\u53d6\u56fe\u50cf\u3001\u6587\u4ef6\u8def\u5f84\u548cnpy\u6587\u4ef6\u8def\u5f84<\/span><br \/>\n        <span class=\"token keyword\">if<\/span> im <span class=\"token keyword\">is<\/span> <span class=\"token boolean\">None<\/span><span class=\"token punctuation\">:<\/span>  <span class=\"token comment\"># \u5982\u679c\u56fe\u50cf\u672a\u7f13\u5b58<\/span><br \/>\n            <span class=\"token keyword\">if<\/span> fn<span class=\"token punctuation\">.<\/span>exists<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>  <span class=\"token comment\"># \u5982\u679cnpy\u6587\u4ef6\u5b58\u5728<\/span><br \/>\n                <span class=\"token keyword\">try<\/span><span class=\"token punctuation\">:<\/span><br \/>\n                    im <span class=\"token operator\">&#061;<\/span> np<span class=\"token punctuation\">.<\/span>load<span class=\"token punctuation\">(<\/span>fn<span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u4ecenpy\u6587\u4ef6\u52a0\u8f7d\u56fe\u50cf<\/span><br \/>\n                <span class=\"token keyword\">except<\/span> Exception <span class=\"token keyword\">as<\/span> e<span class=\"token punctuation\">:<\/span><br \/>\n                    LOGGER<span class=\"token punctuation\">.<\/span>warning<span class=\"token punctuation\">(<\/span><span class=\"token string-interpolation\"><span class=\"token string\">f&#039;<\/span><span class=\"token interpolation\"><span class=\"token punctuation\">{<\/span>self<span class=\"token punctuation\">.<\/span>prefix<span class=\"token punctuation\">}<\/span><\/span><span class=\"token string\">WARNING \u26a0\ufe0f Removing corrupt *.npy image file <\/span><span class=\"token interpolation\"><span class=\"token punctuation\">{<\/span>fn<span class=\"token punctuation\">}<\/span><\/span><span class=\"token string\"> due to: <\/span><span class=\"token interpolation\"><span class=\"token punctuation\">{<\/span>e<span class=\"token punctuation\">}<\/span><\/span><span class=\"token string\">&#039;<\/span><\/span><span class=\"token punctuation\">)<\/span><br \/>\n                    Path<span class=\"token punctuation\">(<\/span>fn<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">.<\/span>unlink<span class=\"token punctuation\">(<\/span>missing_ok<span class=\"token operator\">&#061;<\/span><span class=\"token boolean\">True<\/span><span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u5220\u9664\u635f\u574f\u7684npy\u6587\u4ef6<\/span><br \/>\n                    im <span class=\"token operator\">&#061;<\/span> cv2<span class=\"token punctuation\">.<\/span>imread<span class=\"token punctuation\">(<\/span>f<span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u4ece\u539f\u59cb\u6587\u4ef6\u52a0\u8f7d\u56fe\u50cf<\/span><br \/>\n            <span class=\"token keyword\">else<\/span><span class=\"token punctuation\">:<\/span>  <span class=\"token comment\"># \u4ece\u539f\u59cb\u6587\u4ef6\u52a0\u8f7d\u56fe\u50cf<\/span><br \/>\n                im <span class=\"token operator\">&#061;<\/span> cv2<span class=\"token punctuation\">.<\/span>imread<span class=\"token punctuation\">(<\/span>f<span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># BGR\u683c\u5f0f<\/span><br \/>\n            <span class=\"token keyword\">if<\/span> im <span class=\"token keyword\">is<\/span> <span class=\"token boolean\">None<\/span><span class=\"token punctuation\">:<\/span><br \/>\n                <span class=\"token keyword\">raise<\/span> FileNotFoundError<span class=\"token punctuation\">(<\/span><span class=\"token string-interpolation\"><span class=\"token string\">f&#039;Image Not Found <\/span><span class=\"token interpolation\"><span class=\"token punctuation\">{<\/span>f<span class=\"token punctuation\">}<\/span><\/span><span class=\"token string\">&#039;<\/span><\/span><span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u56fe\u50cf\u672a\u627e\u5230<\/span><\/p>\n<p>            h0<span class=\"token punctuation\">,<\/span> w0 <span class=\"token operator\">&#061;<\/span> im<span class=\"token punctuation\">.<\/span>shape<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token number\">2<\/span><span class=\"token punctuation\">]<\/span>  <span class=\"token comment\"># \u539f\u59cb\u9ad8\u5ea6\u548c\u5bbd\u5ea6<\/span><br \/>\n            <span class=\"token keyword\">if<\/span> rect_mode<span class=\"token punctuation\">:<\/span>  <span class=\"token comment\"># \u5982\u679c\u662f\u77e9\u5f62\u6a21\u5f0f<\/span><br \/>\n                r <span class=\"token operator\">&#061;<\/span> self<span class=\"token punctuation\">.<\/span>imgsz <span class=\"token operator\">\/<\/span> <span class=\"token builtin\">max<\/span><span class=\"token punctuation\">(<\/span>h0<span class=\"token punctuation\">,<\/span> w0<span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u8ba1\u7b97\u7f29\u653e\u6bd4\u4f8b<\/span><br \/>\n                <span class=\"token keyword\">if<\/span> r <span class=\"token operator\">!&#061;<\/span> <span class=\"token number\">1<\/span><span class=\"token punctuation\">:<\/span>  <span class=\"token comment\"># \u5982\u679c\u5927\u5c0f\u4e0d\u76f8\u7b49<\/span><br \/>\n                    w<span class=\"token punctuation\">,<\/span> h <span class=\"token operator\">&#061;<\/span> <span class=\"token punctuation\">(<\/span><span class=\"token builtin\">min<\/span><span class=\"token punctuation\">(<\/span>math<span class=\"token punctuation\">.<\/span>ceil<span class=\"token punctuation\">(<\/span>w0 <span class=\"token operator\">*<\/span> r<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">,<\/span> self<span class=\"token punctuation\">.<\/span>imgsz<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token builtin\">min<\/span><span class=\"token punctuation\">(<\/span>math<span class=\"token punctuation\">.<\/span>ceil<span class=\"token punctuation\">(<\/span>h0 <span class=\"token operator\">*<\/span> r<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">,<\/span> self<span class=\"token punctuation\">.<\/span>imgsz<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u8ba1\u7b97\u65b0\u7684\u5bbd\u9ad8<\/span><br \/>\n                    im <span class=\"token operator\">&#061;<\/span> cv2<span class=\"token punctuation\">.<\/span>resize<span class=\"token punctuation\">(<\/span>im<span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">(<\/span>w<span class=\"token punctuation\">,<\/span> h<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">,<\/span> interpolation<span class=\"token operator\">&#061;<\/span>cv2<span class=\"token punctuation\">.<\/span>INTER_LINEAR<span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u6309\u6bd4\u4f8b\u8c03\u6574\u56fe\u50cf\u5927\u5c0f<\/span><br \/>\n            <span class=\"token keyword\">elif<\/span> <span class=\"token keyword\">not<\/span> <span class=\"token punctuation\">(<\/span>h0 <span class=\"token operator\">&#061;&#061;<\/span> w0 <span class=\"token operator\">&#061;&#061;<\/span> self<span class=\"token punctuation\">.<\/span>imgsz<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>  <span class=\"token comment\"># \u5982\u679c\u4e0d\u662f\u6b63\u65b9\u5f62\u4e14\u4e0d\u7b49\u4e8e\u76ee\u6807\u5927\u5c0f<\/span><br \/>\n                im <span class=\"token operator\">&#061;<\/span> cv2<span class=\"token punctuation\">.<\/span>resize<span class=\"token punctuation\">(<\/span>im<span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">.<\/span>imgsz<span class=\"token punctuation\">,<\/span> self<span class=\"token punctuation\">.<\/span>imgsz<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">,<\/span> interpolation<span class=\"token operator\">&#061;<\/span>cv2<span class=\"token punctuation\">.<\/span>INTER_LINEAR<span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u62c9\u4f38\u56fe\u50cf\u4e3a\u6b63\u65b9\u5f62<\/span><\/p>\n<p>            <span class=\"token comment\"># \u5982\u679c\u8bad\u7ec3\u65f6\u4f7f\u7528\u589e\u5f3a&#xff0c;\u5219\u5c06\u56fe\u50cf\u6dfb\u52a0\u5230\u7f13\u51b2\u533a<\/span><br \/>\n            <span class=\"token keyword\">if<\/span> self<span class=\"token punctuation\">.<\/span>augment<span class=\"token punctuation\">:<\/span><br \/>\n                self<span class=\"token punctuation\">.<\/span>ims<span class=\"token punctuation\">[<\/span>i<span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span> self<span class=\"token punctuation\">.<\/span>im_hw0<span class=\"token punctuation\">[<\/span>i<span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span> self<span class=\"token punctuation\">.<\/span>im_hw<span class=\"token punctuation\">[<\/span>i<span class=\"token punctuation\">]<\/span> <span class=\"token operator\">&#061;<\/span> im<span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">(<\/span>h0<span class=\"token punctuation\">,<\/span> w0<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">,<\/span> im<span class=\"token punctuation\">.<\/span>shape<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token number\">2<\/span><span class=\"token punctuation\">]<\/span>  <span class=\"token comment\"># \u7f13\u5b58\u56fe\u50cf\u548c\u5c3a\u5bf8<\/span><br \/>\n                self<span class=\"token punctuation\">.<\/span><span class=\"token builtin\">buffer<\/span><span class=\"token punctuation\">.<\/span>append<span class=\"token punctuation\">(<\/span>i<span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u5c06\u7d22\u5f15\u6dfb\u52a0\u5230\u7f13\u51b2\u533a<\/span><br \/>\n                <span class=\"token keyword\">if<\/span> <span class=\"token builtin\">len<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">.<\/span><span class=\"token builtin\">buffer<\/span><span class=\"token punctuation\">)<\/span> <span class=\"token operator\">&gt;&#061;<\/span> self<span class=\"token punctuation\">.<\/span>max_buffer_length<span class=\"token punctuation\">:<\/span>  <span class=\"token comment\"># \u5982\u679c\u7f13\u51b2\u533a\u5df2\u6ee1<\/span><br \/>\n                    j <span class=\"token operator\">&#061;<\/span> self<span class=\"token punctuation\">.<\/span><span class=\"token builtin\">buffer<\/span><span class=\"token punctuation\">.<\/span>pop<span class=\"token punctuation\">(<\/span><span class=\"token number\">0<\/span><span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u79fb\u9664\u6700\u65e7\u7684\u7d22\u5f15<\/span><br \/>\n                    self<span class=\"token punctuation\">.<\/span>ims<span class=\"token punctuation\">[<\/span>j<span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span> self<span class=\"token punctuation\">.<\/span>im_hw0<span class=\"token punctuation\">[<\/span>j<span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span> self<span class=\"token punctuation\">.<\/span>im_hw<span class=\"token punctuation\">[<\/span>j<span class=\"token punctuation\">]<\/span> <span class=\"token operator\">&#061;<\/span> <span class=\"token boolean\">None<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token boolean\">None<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token boolean\">None<\/span>  <span class=\"token comment\"># \u6e05\u7a7a\u7f13\u5b58<\/span><\/p>\n<p>            <span class=\"token keyword\">return<\/span> im<span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">(<\/span>h0<span class=\"token punctuation\">,<\/span> w0<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">,<\/span> im<span class=\"token punctuation\">.<\/span>shape<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token number\">2<\/span><span class=\"token punctuation\">]<\/span>  <span class=\"token comment\"># \u8fd4\u56de\u56fe\u50cf\u548c\u5c3a\u5bf8<\/span><\/p>\n<p>        <span class=\"token keyword\">return<\/span> self<span class=\"token punctuation\">.<\/span>ims<span class=\"token punctuation\">[<\/span>i<span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span> self<span class=\"token punctuation\">.<\/span>im_hw0<span class=\"token punctuation\">[<\/span>i<span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span> self<span class=\"token punctuation\">.<\/span>im_hw<span class=\"token punctuation\">[<\/span>i<span class=\"token punctuation\">]<\/span>  <span class=\"token comment\"># \u8fd4\u56de\u7f13\u5b58\u7684\u56fe\u50cf\u548c\u5c3a\u5bf8<\/span><\/p>\n<p>    <span class=\"token keyword\">def<\/span> <span class=\"token function\">__getitem__<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> index<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n        <span class=\"token triple-quoted-string string\">&#034;&#034;&#034;\u8fd4\u56de\u7ed9\u5b9a\u7d22\u5f15\u7684\u8f6c\u6362\u6807\u7b7e\u4fe1\u606f\u3002&#034;&#034;&#034;<\/span><br \/>\n        <span class=\"token keyword\">return<\/span> self<span class=\"token punctuation\">.<\/span>transforms<span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">.<\/span>get_image_and_label<span class=\"token punctuation\">(<\/span>index<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u5e94\u7528\u8f6c\u6362\u5e76\u8fd4\u56de\u56fe\u50cf\u548c\u6807\u7b7e<\/span><\/p>\n<p>    <span class=\"token keyword\">def<\/span> <span class=\"token function\">__len__<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n        <span class=\"token triple-quoted-string string\">&#034;&#034;&#034;\u8fd4\u56de\u6570\u636e\u96c6\u4e2d\u6807\u7b7e\u5217\u8868\u7684\u957f\u5ea6\u3002&#034;&#034;&#034;<\/span><br \/>\n        <span class=\"token keyword\">return<\/span> <span class=\"token builtin\">len<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">.<\/span>labels<span class=\"token punctuation\">)<\/span>  <span class=\"token comment\"># \u8fd4\u56de\u6807\u7b7e\u6570\u91cf<\/span><\/p>\n<p>    <span class=\"token keyword\">def<\/span> <span class=\"token function\">build_transforms<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> hyp<span class=\"token operator\">&#061;<\/span><span class=\"token boolean\">None<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n        <span class=\"token triple-quoted-string string\">&#034;&#034;&#034;<br \/>\n        \u7528\u6237\u53ef\u4ee5\u5728\u6b64\u5904\u81ea\u5b9a\u4e49\u589e\u5f3a\u3002<\/p>\n<p>        \u793a\u4f8b:<br \/>\n            &#096;&#096;&#096;python<br \/>\n            if self.augment:<br \/>\n                # \u8bad\u7ec3\u8f6c\u6362<br \/>\n                return Compose([])<br \/>\n            else:<br \/>\n                # \u9a8c\u8bc1\u8f6c\u6362<br \/>\n                return Compose([])<br \/>\n            &#096;&#096;&#096;<br \/>\n        &#034;&#034;&#034;<\/span><br \/>\n        <span class=\"token keyword\">raise<\/span> NotImplementedError  <span class=\"token comment\"># \u9700\u8981\u7528\u6237\u5b9e\u73b0<\/span><\/p>\n<p>    <span class=\"token keyword\">def<\/span> <span class=\"token function\">get_labels<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n        <span class=\"token triple-quoted-string string\">&#034;&#034;&#034;<br \/>\n        \u7528\u6237\u53ef\u4ee5\u5728\u6b64\u5904\u81ea\u5b9a\u4e49\u6807\u7b7e\u683c\u5f0f\u3002<\/p>\n<p>        \u6ce8\u610f:<br \/>\n            \u786e\u4fdd\u8f93\u51fa\u662f\u4e00\u4e2a\u5b57\u5178&#xff0c;\u5305\u542b\u4ee5\u4e0b\u952e:<br \/>\n            &#096;&#096;&#096;python<br \/>\n            dict(<br \/>\n                im_file&#061;im_file,<br \/>\n                shape&#061;shape,  # \u683c\u5f0f: (\u9ad8\u5ea6, \u5bbd\u5ea6)<br \/>\n                cls&#061;cls,<br \/>\n                bboxes&#061;bboxes, # xywh<br \/>\n                segments&#061;segments,  # xy<br \/>\n                keypoints&#061;keypoints, # xy<br \/>\n                normalized&#061;True, # \u6216 False<br \/>\n                bbox_format&#061;&#034;xyxy&#034;,  # \u6216 xywh, ltwh<br \/>\n            )<br \/>\n            &#096;&#096;&#096;<br \/>\n        &#034;&#034;&#034;<\/span><br \/>\n        <span class=\"token keyword\">raise<\/span> NotImplementedError  <span class=\"token comment\"># \u9700\u8981\u7528\u6237\u5b9e\u73b0<\/span><\/p>\n<h4>\u6838\u5fc3\u90e8\u5206\u5206\u6790<\/h4>\n<li>\n<p>\u521d\u59cb\u5316\u65b9\u6cd5 __init__:<\/p>\n<ul>\n<li>\u8be5\u65b9\u6cd5\u7528\u4e8e\u521d\u59cb\u5316\u6570\u636e\u96c6\u7684\u57fa\u672c\u53c2\u6570&#xff0c;\u5305\u62ec\u56fe\u50cf\u8def\u5f84\u3001\u56fe\u50cf\u5927\u5c0f\u3001\u662f\u5426\u7f13\u5b58\u3001\u6570\u636e\u589e\u5f3a\u7b49\u3002\u5b83\u8fd8\u4f1a\u8c03\u7528\u5176\u4ed6\u65b9\u6cd5\u6765\u83b7\u53d6\u56fe\u50cf\u6587\u4ef6\u548c\u6807\u7b7e&#xff0c;\u5e76\u8bbe\u7f6e\u77e9\u5f62\u8bad\u7ec3\u7684\u53c2\u6570\u3002<\/li>\n<\/ul>\n<\/li>\n<li>\n<p>\u83b7\u53d6\u56fe\u50cf\u6587\u4ef6 get_img_files:<\/p>\n<ul>\n<li>\u8be5\u65b9\u6cd5\u7528\u4e8e\u8bfb\u53d6\u6307\u5b9a\u8def\u5f84\u4e0b\u7684\u6240\u6709\u56fe\u50cf\u6587\u4ef6&#xff0c;\u5e76\u6839\u636e\u7ed9\u5b9a\u7684\u6bd4\u4f8b\u7b5b\u9009\u51fa\u90e8\u5206\u56fe\u50cf\u3002<\/li>\n<\/ul>\n<\/li>\n<li>\n<p>\u52a0\u8f7d\u56fe\u50cf load_image:<\/p>\n<ul>\n<li>\u8be5\u65b9\u6cd5\u7528\u4e8e\u4ece\u6570\u636e\u96c6\u4e2d\u52a0\u8f7d\u5355\u5f20\u56fe\u50cf&#xff0c;\u652f\u6301\u4ece\u7f13\u5b58\u6216\u539f\u59cb\u6587\u4ef6\u52a0\u8f7d&#xff0c;\u5e76\u6839\u636e\u9700\u8981\u8c03\u6574\u56fe\u50cf\u5927\u5c0f\u3002<\/li>\n<\/ul>\n<\/li>\n<li>\n<p>\u83b7\u53d6\u6807\u7b7e get_labels:<\/p>\n<ul>\n<li>\u8be5\u65b9\u6cd5\u7528\u4e8e\u83b7\u53d6\u6807\u7b7e\u4fe1\u606f&#xff0c;\u7528\u6237\u53ef\u4ee5\u81ea\u5b9a\u4e49\u6807\u7b7e\u7684\u683c\u5f0f\u3002<\/li>\n<\/ul>\n<\/li>\n<li>\n<p>\u8f6c\u6362\u65b9\u6cd5 build_transforms:<\/p>\n<ul>\n<li>\u8be5\u65b9\u6cd5\u7528\u4e8e\u6784\u5efa\u56fe\u50cf\u8f6c\u6362&#xff0c;\u7528\u6237\u53ef\u4ee5\u5728\u6b64\u5904\u81ea\u5b9a\u4e49\u6570\u636e\u589e\u5f3a\u7684\u64cd\u4f5c\u3002<\/li>\n<\/ul>\n<\/li>\n<li>\n<p>\u7d22\u5f15\u65b9\u6cd5 __getitem__:<\/p>\n<ul>\n<li>\u8be5\u65b9\u6cd5\u8fd4\u56de\u7ed9\u5b9a\u7d22\u5f15\u7684\u56fe\u50cf\u548c\u6807\u7b7e\u4fe1\u606f&#xff0c;\u5e76\u5e94\u7528\u9884\u5b9a\u4e49\u7684\u8f6c\u6362\u3002<\/li>\n<\/ul>\n<\/li>\n<li>\n<p>\u957f\u5ea6\u65b9\u6cd5 __len__:<\/p>\n<ul>\n<li>\u8be5\u65b9\u6cd5\u8fd4\u56de\u6570\u636e\u96c6\u4e2d\u6807\u7b7e\u7684\u6570\u91cf\u3002<\/li>\n<\/ul>\n<\/li>\n<p>\u901a\u8fc7\u8fd9\u4e9b\u6838\u5fc3\u65b9\u6cd5&#xff0c;BaseDataset \u7c7b\u80fd\u591f\u6709\u6548\u5730\u7ba1\u7406\u56fe\u50cf\u6570\u636e\u7684\u52a0\u8f7d\u3001\u5904\u7406\u548c\u589e\u5f3a&#xff0c;\u4e3a\u540e\u7eed\u7684\u6a21\u578b\u8bad\u7ec3\u63d0\u4f9b\u652f\u6301\u3002&#096;&#096;&#096;<br \/>\n\u8fd9\u4e2a\u6587\u4ef6\u662fUltralytics YOLO\u9879\u76ee\u4e2d\u7684\u4e00\u4e2a\u57fa\u7840\u6570\u636e\u96c6\u7c7bBaseDataset&#xff0c;\u7528\u4e8e\u52a0\u8f7d\u548c\u5904\u7406\u56fe\u50cf\u6570\u636e\u3002\u8be5\u7c7b\u7ee7\u627f\u81eaPyTorch\u7684Dataset&#xff0c;\u4e3b\u8981\u529f\u80fd\u662f\u4e3a\u6a21\u578b\u8bad\u7ec3\u63d0\u4f9b\u56fe\u50cf\u53ca\u5176\u5bf9\u5e94\u7684\u6807\u7b7e\u4fe1\u606f\u3002<\/p>\n<p>\u5728\u521d\u59cb\u5316\u65b9\u6cd5__init__\u4e2d&#xff0c;\u7528\u6237\u53ef\u4ee5\u4f20\u5165\u591a\u4e2a\u53c2\u6570\u6765\u914d\u7f6e\u6570\u636e\u96c6\u7684\u52a0\u8f7d\u65b9\u5f0f&#xff0c;\u5305\u62ec\u56fe\u50cf\u8def\u5f84\u3001\u56fe\u50cf\u5927\u5c0f\u3001\u662f\u5426\u7f13\u5b58\u56fe\u50cf\u3001\u6570\u636e\u589e\u5f3a\u9009\u9879\u3001\u8d85\u53c2\u6570\u3001\u6279\u91cf\u5927\u5c0f\u7b49\u3002\u521d\u59cb\u5316\u8fc7\u7a0b\u4e2d&#xff0c;\u7c7b\u4f1a\u8c03\u7528get_img_files\u65b9\u6cd5\u83b7\u53d6\u56fe\u50cf\u6587\u4ef6\u5217\u8868&#xff0c;\u5e76\u901a\u8fc7get_labels\u65b9\u6cd5\u83b7\u53d6\u6807\u7b7e\u4fe1\u606f\u3002\u540c\u65f6&#xff0c;\u5b83\u8fd8\u4f1a\u6839\u636e\u7528\u6237\u7684\u8bbe\u7f6e\u8fdb\u884c\u56fe\u50cf\u7f13\u5b58\u548c\u6570\u636e\u589e\u5f3a\u7684\u51c6\u5907\u3002<\/p>\n<p>get_img_files\u65b9\u6cd5\u8d1f\u8d23\u8bfb\u53d6\u6307\u5b9a\u8def\u5f84\u4e0b\u7684\u56fe\u50cf\u6587\u4ef6\u3002\u5b83\u652f\u6301\u4f20\u5165\u76ee\u5f55\u6216\u6587\u4ef6\u5217\u8868&#xff0c;\u5e76\u80fd\u591f\u9012\u5f52\u67e5\u627e\u56fe\u50cf\u6587\u4ef6\u3002\u8be5\u65b9\u6cd5\u4f1a\u786e\u4fdd\u8fd4\u56de\u7684\u6587\u4ef6\u662f\u6709\u6548\u7684\u56fe\u50cf\u683c\u5f0f&#xff0c;\u5e76\u6839\u636e\u9700\u8981\u9650\u5236\u4f7f\u7528\u7684\u6570\u636e\u96c6\u5927\u5c0f\u3002<\/p>\n<p>update_labels\u65b9\u6cd5\u7528\u4e8e\u66f4\u65b0\u6807\u7b7e&#xff0c;\u53ea\u4fdd\u7559\u7528\u6237\u6307\u5b9a\u7684\u7c7b\u522b\u3002\u5b83\u4f1a\u6839\u636e\u4f20\u5165\u7684\u7c7b\u522b\u5217\u8868\u7b5b\u9009\u6807\u7b7e\u4fe1\u606f&#xff0c;\u5e76\u5904\u7406\u5355\u7c7b\u8bad\u7ec3\u7684\u60c5\u51b5\u3002<\/p>\n<p>load_image\u65b9\u6cd5\u8d1f\u8d23\u52a0\u8f7d\u6307\u5b9a\u7d22\u5f15\u7684\u56fe\u50cf&#xff0c;\u5e76\u6839\u636e\u9700\u8981\u8fdb\u884c\u8c03\u6574&#xff08;\u5982\u7f29\u653e&#xff09;\u3002\u5982\u679c\u56fe\u50cf\u5df2\u7ecf\u7f13\u5b58&#xff0c;\u5219\u76f4\u63a5\u8fd4\u56de\u7f13\u5b58\u7684\u56fe\u50cf&#xff1b;\u5426\u5219&#xff0c;\u5b83\u4f1a\u5c1d\u8bd5\u4ece\u78c1\u76d8\u8bfb\u53d6\u56fe\u50cf&#xff0c;\u5e76\u5728\u5fc5\u8981\u65f6\u8fdb\u884c\u5c3a\u5bf8\u8c03\u6574\u3002<\/p>\n<p>cache_images\u548ccache_images_to_disk\u65b9\u6cd5\u5219\u7528\u4e8e\u5c06\u56fe\u50cf\u7f13\u5b58\u5230\u5185\u5b58\u6216\u78c1\u76d8&#xff0c;\u4ee5\u63d0\u9ad8\u540e\u7eed\u52a0\u8f7d\u7684\u901f\u5ea6\u3002check_cache_ram\u65b9\u6cd5\u4f1a\u68c0\u67e5\u5f53\u524d\u7cfb\u7edf\u7684\u5185\u5b58\u662f\u5426\u8db3\u591f\u8fdb\u884c\u56fe\u50cf\u7f13\u5b58\u3002<\/p>\n<p>set_rectangle\u65b9\u6cd5\u7528\u4e8e\u8bbe\u7f6eYOLO\u68c0\u6d4b\u7684\u8fb9\u754c\u6846\u5f62\u72b6\u4e3a\u77e9\u5f62&#xff0c;\u4ee5\u9002\u5e94\u4e0d\u540c\u7684\u56fe\u50cf\u957f\u5bbd\u6bd4\u3002\u5b83\u4f1a\u6839\u636e\u6279\u6b21\u5927\u5c0f\u548c\u56fe\u50cf\u7684\u957f\u5bbd\u6bd4\u8c03\u6574\u56fe\u50cf\u7684\u987a\u5e8f\u548c\u5f62\u72b6\u3002<\/p>\n<p>__getitem__\u65b9\u6cd5\u662fPyTorch\u6570\u636e\u96c6\u7684\u6807\u51c6\u63a5\u53e3&#xff0c;\u7528\u4e8e\u8fd4\u56de\u7ed9\u5b9a\u7d22\u5f15\u7684\u56fe\u50cf\u548c\u6807\u7b7e\u4fe1\u606f&#xff0c;\u5e76\u5e94\u7528\u76f8\u5e94\u7684\u53d8\u6362\u3002get_image_and_label\u65b9\u6cd5\u5219\u7528\u4e8e\u83b7\u53d6\u56fe\u50cf\u53ca\u5176\u6807\u7b7e\u7684\u8be6\u7ec6\u4fe1\u606f\u3002<\/p>\n<p>__len__\u65b9\u6cd5\u8fd4\u56de\u6570\u636e\u96c6\u4e2d\u6807\u7b7e\u7684\u6570\u91cf&#xff0c;\u800cupdate_labels_info\u65b9\u6cd5\u5219\u5141\u8bb8\u7528\u6237\u81ea\u5b9a\u4e49\u6807\u7b7e\u683c\u5f0f\u3002<\/p>\n<p>\u6700\u540e&#xff0c;build_transforms\u548cget_labels\u65b9\u6cd5\u662f\u7559\u7ed9\u7528\u6237\u81ea\u5b9a\u4e49\u7684\u6570\u636e\u589e\u5f3a\u548c\u6807\u7b7e\u683c\u5f0f\u5316\u7684\u63a5\u53e3&#xff0c;\u7528\u6237\u53ef\u4ee5\u6839\u636e\u5177\u4f53\u9700\u6c42\u5b9e\u73b0\u8fd9\u4e9b\u65b9\u6cd5\u3002<\/p>\n<p>\u603b\u4f53\u6765\u8bf4&#xff0c;\u8fd9\u4e2a\u6587\u4ef6\u63d0\u4f9b\u4e86\u4e00\u4e2a\u7075\u6d3b\u7684\u6846\u67b6&#xff0c;\u7528\u4e8e\u5904\u7406\u56fe\u50cf\u6570\u636e\u96c6&#xff0c;\u652f\u6301\u591a\u79cd\u914d\u7f6e\u9009\u9879&#xff0c;\u9002\u5408\u7528\u4e8eYOLO\u6a21\u578b\u7684\u8bad\u7ec3\u548c\u8bc4\u4f30\u3002<\/p>\n<p><span class=\"token keyword\">import<\/span> sys<br \/>\n<span class=\"token keyword\">import<\/span> subprocess<\/p>\n<p><span class=\"token keyword\">def<\/span> <span class=\"token function\">run_script<\/span><span class=\"token punctuation\">(<\/span>script_path<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span><br \/>\n    <span class=\"token triple-quoted-string string\">&#034;&#034;&#034;<br \/>\n    \u4f7f\u7528\u5f53\u524d Python \u73af\u5883\u8fd0\u884c\u6307\u5b9a\u7684\u811a\u672c\u3002<\/p>\n<p>    Args:<br \/>\n        script_path (str): \u8981\u8fd0\u884c\u7684\u811a\u672c\u8def\u5f84<\/p>\n<p>    Returns:<br \/>\n        None<br \/>\n    &#034;&#034;&#034;<\/span><br \/>\n    <span class=\"token comment\"># \u83b7\u53d6\u5f53\u524d Python \u89e3\u91ca\u5668\u7684\u8def\u5f84<\/span><br \/>\n    python_path <span class=\"token operator\">&#061;<\/span> sys<span class=\"token punctuation\">.<\/span>executable<\/p>\n<p>    <span class=\"token comment\"># \u6784\u5efa\u8fd0\u884c\u547d\u4ee4&#xff0c;\u4f7f\u7528 streamlit \u8fd0\u884c\u6307\u5b9a\u7684\u811a\u672c<\/span><br \/>\n    command <span class=\"token operator\">&#061;<\/span> <span class=\"token string-interpolation\"><span class=\"token string\">f&#039;&#034;<\/span><span class=\"token interpolation\"><span class=\"token punctuation\">{<\/span>python_path<span class=\"token punctuation\">}<\/span><\/span><span class=\"token string\">&#034; -m streamlit run &#034;<\/span><span class=\"token interpolation\"><span class=\"token punctuation\">{<\/span>script_path<span class=\"token punctuation\">}<\/span><\/span><span class=\"token string\">&#034;&#039;<\/span><\/span><\/p>\n<p>    <span class=\"token comment\"># \u6267\u884c\u547d\u4ee4\u5e76\u7b49\u5f85\u5176\u5b8c\u6210<\/span><br \/>\n    result <span class=\"token operator\">&#061;<\/span> subprocess<span class=\"token punctuation\">.<\/span>run<span class=\"token punctuation\">(<\/span>command<span class=\"token punctuation\">,<\/span> shell<span class=\"token operator\">&#061;<\/span><span class=\"token boolean\">True<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<p>    <span class=\"token comment\"># \u68c0\u67e5\u547d\u4ee4\u6267\u884c\u7ed3\u679c&#xff0c;\u5982\u679c\u8fd4\u56de\u7801\u4e0d\u4e3a0&#xff0c;\u8868\u793a\u6267\u884c\u51fa\u9519<\/span><br \/>\n    <span class=\"token keyword\">if<\/span> result<span class=\"token punctuation\">.<\/span>returncode <span class=\"token operator\">!&#061;<\/span> <span class=\"token number\">0<\/span><span class=\"token punctuation\">:<\/span><br \/>\n        <span class=\"token keyword\">print<\/span><span class=\"token punctuation\">(<\/span><span class=\"token string\">&#034;\u811a\u672c\u8fd0\u884c\u51fa\u9519\u3002&#034;<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<p><span class=\"token comment\"># \u4e3b\u7a0b\u5e8f\u5165\u53e3<\/span><br \/>\n<span class=\"token keyword\">if<\/span> __name__ <span class=\"token operator\">&#061;&#061;<\/span> <span class=\"token string\">&#034;__main__&#034;<\/span><span class=\"token punctuation\">:<\/span><br \/>\n    <span class=\"token comment\"># \u6307\u5b9a\u8981\u8fd0\u884c\u7684\u811a\u672c\u8def\u5f84<\/span><br \/>\n    script_path <span class=\"token operator\">&#061;<\/span> <span class=\"token string\">&#034;web.py&#034;<\/span>  <span class=\"token comment\"># \u5047\u8bbe\u811a\u672c\u5728\u5f53\u524d\u76ee\u5f55\u4e0b<\/span><\/p>\n<p>    <span class=\"token comment\"># \u8c03\u7528\u51fd\u6570\u8fd0\u884c\u811a\u672c<\/span><br \/>\n    run_script<span class=\"token punctuation\">(<\/span>script_path<span class=\"token punctuation\">)<\/span><\/p>\n<h4>\u4ee3\u7801\u6ce8\u91ca\u8bf4\u660e&#xff1a;<\/h4>\n<li>\n<p>\u5bfc\u5165\u6a21\u5757&#xff1a;<\/p>\n<ul>\n<li>sys&#xff1a;\u7528\u4e8e\u8bbf\u95ee\u4e0e Python \u89e3\u91ca\u5668\u76f8\u5173\u7684\u53d8\u91cf\u548c\u51fd\u6570\u3002<\/li>\n<li>subprocess&#xff1a;\u7528\u4e8e\u6267\u884c\u5916\u90e8\u547d\u4ee4\u548c\u7a0b\u5e8f\u3002<\/li>\n<\/ul>\n<\/li>\n<li>\n<p>run_script \u51fd\u6570&#xff1a;<\/p>\n<ul>\n<li>\u63a5\u53d7\u4e00\u4e2a\u53c2\u6570 script_path&#xff0c;\u8868\u793a\u8981\u8fd0\u884c\u7684 Python \u811a\u672c\u7684\u8def\u5f84\u3002<\/li>\n<li>\u4f7f\u7528 sys.executable \u83b7\u53d6\u5f53\u524d Python \u89e3\u91ca\u5668\u7684\u8def\u5f84&#xff0c;\u4ee5\u786e\u4fdd\u811a\u672c\u5728\u5f53\u524d\u73af\u5883\u4e2d\u8fd0\u884c\u3002<\/li>\n<li>\u6784\u5efa\u547d\u4ee4\u5b57\u7b26\u4e32&#xff0c;\u4f7f\u7528 streamlit \u6a21\u5757\u8fd0\u884c\u6307\u5b9a\u7684\u811a\u672c\u3002<\/li>\n<li>\u4f7f\u7528 subprocess.run \u6267\u884c\u547d\u4ee4&#xff0c;\u5e76\u7b49\u5f85\u5176\u5b8c\u6210\u3002<\/li>\n<li>\u68c0\u67e5\u547d\u4ee4\u7684\u8fd4\u56de\u7801&#xff0c;\u5982\u679c\u4e0d\u4e3a0&#xff0c;\u8868\u793a\u6267\u884c\u8fc7\u7a0b\u4e2d\u51fa\u73b0\u9519\u8bef&#xff0c;\u6253\u5370\u9519\u8bef\u4fe1\u606f\u3002<\/li>\n<\/ul>\n<\/li>\n<li>\n<p>\u4e3b\u7a0b\u5e8f\u5165\u53e3&#xff1a;<\/p>\n<ul>\n<li>\u4f7f\u7528 if __name__ &#061;&#061; &#034;__main__&#034;: \u786e\u4fdd\u4ee3\u7801\u5757\u53ea\u5728\u76f4\u63a5\u8fd0\u884c\u8be5\u811a\u672c\u65f6\u6267\u884c\u3002<\/li>\n<li>\u6307\u5b9a\u8981\u8fd0\u884c\u7684\u811a\u672c\u8def\u5f84&#xff08;\u5728\u6b64\u793a\u4f8b\u4e2d\u4e3a web.py&#xff09;\u3002<\/li>\n<li>\u8c03\u7528 run_script \u51fd\u6570&#xff0c;\u4f20\u5165\u811a\u672c\u8def\u5f84\u4ee5\u6267\u884c\u8be5\u811a\u672c\u3002&#096;&#096;&#096;<br \/>\n\u8fd9\u4e2a\u7a0b\u5e8f\u6587\u4ef6\u7684\u4e3b\u8981\u529f\u80fd\u662f\u901a\u8fc7\u5f53\u524d\u7684 Python \u73af\u5883\u6765\u8fd0\u884c\u4e00\u4e2a\u6307\u5b9a\u7684\u811a\u672c&#xff0c;\u5177\u4f53\u662f\u4f7f\u7528 Streamlit \u6765\u542f\u52a8\u4e00\u4e2a web \u5e94\u7528\u3002\u9996\u5148&#xff0c;\u7a0b\u5e8f\u5bfc\u5165\u4e86\u5fc5\u8981\u7684\u6a21\u5757&#xff0c;\u5305\u62ec sys\u3001os \u548c subprocess&#xff0c;\u8fd9\u4e9b\u6a21\u5757\u63d0\u4f9b\u4e86\u4e0e\u7cfb\u7edf\u4ea4\u4e92\u7684\u529f\u80fd\u3002abs_path \u51fd\u6570\u5219\u7528\u4e8e\u83b7\u53d6\u6587\u4ef6\u7684\u7edd\u5bf9\u8def\u5f84\u3002<\/li>\n<\/ul>\n<\/li>\n<p>\u5728 run_script \u51fd\u6570\u4e2d&#xff0c;\u9996\u5148\u83b7\u53d6\u5f53\u524d Python \u89e3\u91ca\u5668\u7684\u8def\u5f84&#xff0c;\u8fd9\u6837\u53ef\u4ee5\u786e\u4fdd\u4f7f\u7528\u6b63\u786e\u7684 Python \u73af\u5883\u6765\u6267\u884c\u811a\u672c\u3002\u63a5\u7740&#xff0c;\u6784\u5efa\u4e00\u4e2a\u547d\u4ee4\u5b57\u7b26\u4e32&#xff0c;\u8fd9\u4e2a\u547d\u4ee4\u4f1a\u8c03\u7528 Streamlit \u6765\u8fd0\u884c\u6307\u5b9a\u7684\u811a\u672c\u3002\u547d\u4ee4\u7684\u683c\u5f0f\u662f\u5c06 Python \u89e3\u91ca\u5668\u8def\u5f84\u4e0e\u8981\u8fd0\u884c\u7684\u811a\u672c\u8def\u5f84\u7ed3\u5408\u8d77\u6765&#xff0c;\u5f62\u6210\u4e00\u4e2a\u5b8c\u6574\u7684\u547d\u4ee4\u3002<\/p>\n<p>\u7136\u540e&#xff0c;\u4f7f\u7528 subprocess.run \u65b9\u6cd5\u6765\u6267\u884c\u8fd9\u4e2a\u547d\u4ee4\u3002\u8be5\u65b9\u6cd5\u4f1a\u5728\u4e00\u4e2a\u65b0\u7684 shell \u4e2d\u8fd0\u884c\u547d\u4ee4&#xff0c;\u5e76\u7b49\u5f85\u5176\u5b8c\u6210\u3002\u5982\u679c\u811a\u672c\u8fd0\u884c\u8fc7\u7a0b\u4e2d\u51fa\u73b0\u9519\u8bef&#xff0c;\u8fd4\u56de\u7684 returncode \u5c06\u4e0d\u4e3a\u96f6&#xff0c;\u6b64\u65f6\u7a0b\u5e8f\u4f1a\u6253\u5370\u51fa\u201c\u811a\u672c\u8fd0\u884c\u51fa\u9519\u201d\u7684\u63d0\u793a\u4fe1\u606f\u3002<\/p>\n<p>\u5728\u6587\u4ef6\u7684\u6700\u540e\u90e8\u5206&#xff0c;\u4f7f\u7528 if __name__ &#061;&#061; &#034;__main__&#034;: \u8bed\u53e5\u6765\u786e\u4fdd\u53ea\u6709\u5728\u76f4\u63a5\u8fd0\u884c\u8be5\u6587\u4ef6\u65f6\u624d\u4f1a\u6267\u884c\u4e0b\u9762\u7684\u4ee3\u7801\u3002\u8fd9\u91cc\u6307\u5b9a\u4e86\u8981\u8fd0\u884c\u7684\u811a\u672c\u8def\u5f84&#xff0c;\u5373 web.py&#xff0c;\u5e76\u8c03\u7528 run_script \u51fd\u6570\u6765\u542f\u52a8\u8fd9\u4e2a\u811a\u672c\u3002<\/p>\n<p>\u603b\u4f53\u6765\u8bf4&#xff0c;\u8fd9\u4e2a\u7a0b\u5e8f\u6587\u4ef6\u7684\u8bbe\u8ba1\u76ee\u7684\u662f\u4e3a\u4e86\u65b9\u4fbf\u5730\u901a\u8fc7 Python \u73af\u5883\u8fd0\u884c\u4e00\u4e2a Streamlit \u5e94\u7528&#xff0c;\u7b80\u5316\u4e86\u542f\u52a8\u8fc7\u7a0b\u3002<\/p>\n<p>&#096;&#096;&#096;python<br \/>\ndef coco91_to_coco80_class():<br \/>\n    &#034;&#034;&#034;<br \/>\n    \u5c0691\u4e2aCOCO\u7c7bID\u8f6c\u6362\u4e3a80\u4e2aCOCO\u7c7bID\u3002<\/p>\n<p>    \u8fd4\u56de:<br \/>\n        (list): \u4e00\u4e2a\u5305\u542b91\u4e2a\u7c7bID\u7684\u5217\u8868&#xff0c;\u5176\u4e2d\u7d22\u5f15\u8868\u793a80\u4e2a\u7c7bID&#xff0c;\u503c\u4e3a\u5bf9\u5e94\u768491\u4e2a\u7c7bID\u3002<br \/>\n    &#034;&#034;&#034;<br \/>\n    return [<br \/>\n        0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, None, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, None, 24, 25, None,<br \/>\n        None, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, None, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50,<br \/>\n        51, 52, 53, 54, 55, 56, 57, 58, 59, None, 60, None, None, 61, None, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72,<br \/>\n        None, 73, 74, 75, 76, 77, 78, 79, None]<\/p>\n<p>def convert_coco(labels_dir&#061;&#039;..\/coco\/annotations\/&#039;,<br \/>\n                 save_dir&#061;&#039;coco_converted\/&#039;,<br \/>\n                 use_segments&#061;False,<br \/>\n                 use_keypoints&#061;False,<br \/>\n                 cls91to80&#061;True):<br \/>\n    &#034;&#034;&#034;<br \/>\n    \u5c06COCO\u6570\u636e\u96c6\u7684\u6ce8\u91ca\u8f6c\u6362\u4e3a\u9002\u5408\u8bad\u7ec3YOLO\u6a21\u578b\u7684YOLO\u6ce8\u91ca\u683c\u5f0f\u3002<\/p>\n<p>    \u53c2\u6570:<br \/>\n        labels_dir (str, optional): \u5305\u542bCOCO\u6570\u636e\u96c6\u6ce8\u91ca\u6587\u4ef6\u7684\u76ee\u5f55\u8def\u5f84\u3002<br \/>\n        save_dir (str, optional): \u4fdd\u5b58\u7ed3\u679c\u7684\u76ee\u5f55\u8def\u5f84\u3002<br \/>\n        use_segments (bool, optional): \u662f\u5426\u5728\u8f93\u51fa\u4e2d\u5305\u542b\u5206\u5272\u63a9\u7801\u3002<br \/>\n        use_keypoints (bool, optional): \u662f\u5426\u5728\u8f93\u51fa\u4e2d\u5305\u542b\u5173\u952e\u70b9\u6ce8\u91ca\u3002<br \/>\n        cls91to80 (bool, optional): \u662f\u5426\u5c0691\u4e2aCOCO\u7c7bID\u6620\u5c04\u5230\u5bf9\u5e94\u768480\u4e2aCOCO\u7c7bID\u3002<\/p>\n<p>    \u8f93\u51fa:<br \/>\n        \u5728\u6307\u5b9a\u7684\u8f93\u51fa\u76ee\u5f55\u4e2d\u751f\u6210\u8f93\u51fa\u6587\u4ef6\u3002<br \/>\n    &#034;&#034;&#034;<\/p>\n<p>    # \u521b\u5efa\u6570\u636e\u96c6\u76ee\u5f55<br \/>\n    save_dir &#061; increment_path(save_dir)  # \u5982\u679c\u4fdd\u5b58\u76ee\u5f55\u5df2\u5b58\u5728&#xff0c;\u5219\u589e\u52a0\u540e\u7f00<br \/>\n    for p in save_dir \/ &#039;labels&#039;, save_dir \/ &#039;images&#039;:<br \/>\n        p.mkdir(parents&#061;True, exist_ok&#061;True)  # \u521b\u5efa\u76ee\u5f55<\/p>\n<p>    # \u8f6c\u6362\u7c7b<br \/>\n    coco80 &#061; coco91_to_coco80_class()<\/p>\n<p>    # \u5bfc\u5165json\u6587\u4ef6<br \/>\n    for json_file in sorted(Path(labels_dir).resolve().glob(&#039;*.json&#039;)):<br \/>\n        fn &#061; Path(save_dir) \/ &#039;labels&#039; \/ json_file.stem.replace(&#039;instances_&#039;, &#039;&#039;)  # \u6587\u4ef6\u5939\u540d\u79f0<br \/>\n        fn.mkdir(parents&#061;True, exist_ok&#061;True)<br \/>\n        with open(json_file) as f:<br \/>\n            data &#061; json.load(f)<\/p>\n<p>        # \u521b\u5efa\u56fe\u50cf\u5b57\u5178<br \/>\n        images &#061; {f&#039;{x[&#034;id&#034;]:d}&#039;: x for x in data[&#039;images&#039;]}<br \/>\n        # \u521b\u5efa\u56fe\u50cf-\u6ce8\u91ca\u5b57\u5178<br \/>\n        imgToAnns &#061; defaultdict(list)<br \/>\n        for ann in data[&#039;annotations&#039;]:<br \/>\n            imgToAnns[ann[&#039;image_id&#039;]].append(ann)<\/p>\n<p>        # \u5199\u5165\u6807\u7b7e\u6587\u4ef6<br \/>\n        for img_id, anns in TQDM(imgToAnns.items(), desc&#061;f&#039;Annotations {json_file}&#039;):<br \/>\n            img &#061; images[f&#039;{img_id:d}&#039;]<br \/>\n            h, w, f &#061; img[&#039;height&#039;], img[&#039;width&#039;], img[&#039;file_name&#039;]<\/p>\n<p>            bboxes &#061; []  # \u5b58\u50a8\u8fb9\u754c\u6846<br \/>\n            segments &#061; []  # \u5b58\u50a8\u5206\u5272<br \/>\n            keypoints &#061; []  # \u5b58\u50a8\u5173\u952e\u70b9<br \/>\n            for ann in anns:<br \/>\n                if ann[&#039;iscrowd&#039;]:<br \/>\n                    continue  # \u8df3\u8fc7\u62e5\u6324\u7684\u6ce8\u91ca<br \/>\n                # COCO\u6846\u683c\u5f0f\u4e3a [\u5de6\u4e0a\u89d2x, \u5de6\u4e0a\u89d2y, \u5bbd\u5ea6, \u9ad8\u5ea6]<br \/>\n                box &#061; np.array(ann[&#039;bbox&#039;], dtype&#061;np.float64)<br \/>\n                box[:2] &#043;&#061; box[2:] \/ 2  # \u5c06\u5de6\u4e0a\u89d2\u5750\u6807\u8f6c\u6362\u4e3a\u4e2d\u5fc3\u5750\u6807<br \/>\n                box[[0, 2]] \/&#061; w  # \u5f52\u4e00\u5316x\u5750\u6807<br \/>\n                box[[1, 3]] \/&#061; h  # \u5f52\u4e00\u5316y\u5750\u6807<br \/>\n                if box[2] &lt;&#061; 0 or box[3] &lt;&#061; 0:  # \u5982\u679c\u5bbd\u5ea6\u6216\u9ad8\u5ea6\u5c0f\u4e8e\u7b49\u4e8e0<br \/>\n                    continue<\/p>\n<p>                cls &#061; coco80[ann[&#039;category_id&#039;] &#8211; 1] if cls91to80 else ann[&#039;category_id&#039;] &#8211; 1  # \u7c7b\u522b<br \/>\n                box &#061; [cls] &#043; box.tolist()  # \u5c06\u7c7b\u522b\u548c\u8fb9\u754c\u6846\u5408\u5e76<br \/>\n                if box not in bboxes:<br \/>\n                    bboxes.append(box)  # \u6dfb\u52a0\u8fb9\u754c\u6846<br \/>\n                if use_segments and ann.get(&#039;segmentation&#039;) is not None:<br \/>\n                    # \u5904\u7406\u5206\u5272<br \/>\n                    # \u7701\u7565\u5206\u5272\u5904\u7406\u4ee3\u7801<br \/>\n                if use_keypoints and ann.get(&#039;keypoints&#039;) is not None:<br \/>\n                    # \u5904\u7406\u5173\u952e\u70b9<br \/>\n                    # \u7701\u7565\u5173\u952e\u70b9\u5904\u7406\u4ee3\u7801<\/p>\n<p>            # \u5199\u5165\u6587\u4ef6<br \/>\n            with open((fn \/ f).with_suffix(&#039;.txt&#039;), &#039;a&#039;) as file:<br \/>\n                for i in range(len(bboxes)):<br \/>\n                    if use_keypoints:<br \/>\n                        line &#061; *(keypoints[i]),  # \u7c7b\u522b, \u8fb9\u754c\u6846, \u5173\u952e\u70b9<br \/>\n                    else:<br \/>\n                        line &#061; *(segments[i] if use_segments and len(segments[i]) &gt; 0 else bboxes[i]),  # \u7c7b\u522b, \u8fb9\u754c\u6846\u6216\u5206\u5272<br \/>\n                    file.write((&#039;%g &#039; * len(line)).rstrip() % line &#043; &#039;\\\\n&#039;)<\/p>\n<p>    LOGGER.info(f&#039;COCO\u6570\u636e\u6210\u529f\u8f6c\u6362\u3002\\\\n\u7ed3\u679c\u4fdd\u5b58\u5230 {save_dir.resolve()}&#039;)<\/p>\n<h4>\u4ee3\u7801\u6838\u5fc3\u90e8\u5206\u89e3\u91ca&#xff1a;<\/h4>\n<li>\n<p>coco91_to_coco80_class: \u8be5\u51fd\u6570\u7528\u4e8e\u5c06COCO\u6570\u636e\u96c6\u4e2d91\u4e2a\u7c7b\u7684ID\u8f6c\u6362\u4e3a80\u4e2a\u7c7b\u7684ID&#xff0c;\u8fd4\u56de\u4e00\u4e2a\u5217\u8868&#xff0c;\u5176\u4e2d\u7d22\u5f15\u5bf9\u5e9480\u4e2a\u7c7b\u7684ID&#xff0c;\u503c\u4e3a\u5bf9\u5e94\u768491\u4e2a\u7c7b\u7684ID\u3002<\/p>\n<\/li>\n<li>\n<p>convert_coco: \u8be5\u51fd\u6570\u8d1f\u8d23\u5c06COCO\u6570\u636e\u96c6\u7684\u6ce8\u91ca\u8f6c\u6362\u4e3aYOLO\u683c\u5f0f\u3002\u5b83\u63a5\u53d7\u591a\u4e2a\u53c2\u6570\u4ee5\u63a7\u5236\u8f6c\u6362\u7684\u7ec6\u8282&#xff0c;\u5305\u62ec\u8f93\u5165\u548c\u8f93\u51fa\u76ee\u5f55\u3001\u662f\u5426\u4f7f\u7528\u5206\u5272\u548c\u5173\u952e\u70b9\u7b49\u3002\u51fd\u6570\u5185\u90e8\u4f1a\u521b\u5efa\u5fc5\u8981\u7684\u76ee\u5f55&#xff0c;\u8bfb\u53d6JSON\u6ce8\u91ca\u6587\u4ef6&#xff0c;\u5904\u7406\u6bcf\u4e2a\u56fe\u50cf\u7684\u6ce8\u91ca\u5e76\u751f\u6210\u76f8\u5e94\u7684YOLO\u683c\u5f0f\u6807\u7b7e\u6587\u4ef6\u3002<\/p>\n<\/li>\n<li>\n<p>\u6ce8\u91ca\u5904\u7406: \u5728\u5904\u7406\u6bcf\u4e2a\u6ce8\u91ca\u65f6&#xff0c;\u51fd\u6570\u4f1a\u6839\u636e\u662f\u5426\u662f\u62e5\u6324\u7684\u6ce8\u91ca\u8fdb\u884c\u8fc7\u6ee4&#xff0c;\u8f6c\u6362\u8fb9\u754c\u6846\u683c\u5f0f&#xff0c;\u5e76\u8fdb\u884c\u5f52\u4e00\u5316\u5904\u7406\u3002\u540c\u65f6&#xff0c;\u6839\u636e\u7528\u6237\u7684\u9009\u62e9&#xff0c;\u53ef\u80fd\u4f1a\u5904\u7406\u5206\u5272\u548c\u5173\u952e\u70b9\u4fe1\u606f\u3002<\/p>\n<\/li>\n<li>\n<p>\u6587\u4ef6\u5199\u5165: \u6700\u540e&#xff0c;\u51fd\u6570\u5c06\u5904\u7406\u540e\u7684\u8fb9\u754c\u6846\u3001\u5206\u5272\u6216\u5173\u952e\u70b9\u4fe1\u606f\u5199\u5165\u5230\u6307\u5b9a\u7684\u6587\u672c\u6587\u4ef6\u4e2d&#xff0c;\u4ee5\u4f9bYOLO\u6a21\u578b\u8bad\u7ec3\u4f7f\u7528\u3002&#096;&#096;&#096;<br \/>\n\u8fd9\u4e2a\u7a0b\u5e8f\u6587\u4ef6\u4e3b\u8981\u7528\u4e8e\u5c06COCO\u6570\u636e\u96c6\u7684\u6807\u6ce8\u683c\u5f0f\u8f6c\u6362\u4e3aYOLO\u6a21\u578b\u6240\u9700\u7684\u6807\u6ce8\u683c\u5f0f\u3002\u6587\u4ef6\u4e2d\u5305\u542b\u591a\u4e2a\u51fd\u6570&#xff0c;\u5177\u4f53\u529f\u80fd\u5982\u4e0b&#xff1a;<\/p>\n<\/li>\n<p>\u9996\u5148&#xff0c;coco91_to_coco80_class\u548ccoco80_to_coco91_class\u4e24\u4e2a\u51fd\u6570\u7528\u4e8e\u5728COCO\u6570\u636e\u96c6\u4e2d\u8fdb\u884c\u7c7b\u522bID\u7684\u8f6c\u6362\u3002COCO\u6570\u636e\u96c6\u539f\u672c\u670991\u4e2a\u7c7b\u522b&#xff0c;\u4f46YOLO\u6a21\u578b\u53ea\u4f7f\u7528\u5176\u4e2d\u768480\u4e2a\u7c7b\u522b\u3002\u8fd9\u4e24\u4e2a\u51fd\u6570\u5206\u522b\u5b9e\u73b0\u4e8691\u523080\u7684\u6620\u5c04\u548c80\u523091\u7684\u6620\u5c04&#xff0c;\u8fd4\u56de\u4e00\u4e2a\u5217\u8868&#xff0c;\u5176\u4e2d\u6bcf\u4e2a\u7d22\u5f15\u5bf9\u5e94\u4e8e\u4e00\u4e2a\u7c7b\u522bID\u3002<\/p>\n<p>\u63a5\u4e0b\u6765&#xff0c;convert_coco\u51fd\u6570\u662f\u6587\u4ef6\u7684\u6838\u5fc3\u529f\u80fd\u3002\u5b83\u63a5\u53d7\u591a\u4e2a\u53c2\u6570&#xff0c;\u5305\u62ec\u6807\u6ce8\u6587\u4ef6\u7684\u76ee\u5f55\u3001\u4fdd\u5b58\u7ed3\u679c\u7684\u76ee\u5f55\u3001\u662f\u5426\u4f7f\u7528\u5206\u5272\u63a9\u7801\u548c\u5173\u952e\u70b9\u7b49\u3002\u8be5\u51fd\u6570\u9996\u5148\u521b\u5efa\u4fdd\u5b58\u7ed3\u679c\u7684\u76ee\u5f55&#xff0c;\u5e76\u5c06COCO\u7c7b\u522b\u8f6c\u6362\u4e3aYOLO\u6240\u9700\u7684\u683c\u5f0f\u3002\u7136\u540e&#xff0c;\u5b83\u904d\u5386\u6307\u5b9a\u76ee\u5f55\u4e0b\u7684\u6240\u6709JSON\u6807\u6ce8\u6587\u4ef6&#xff0c;\u8bfb\u53d6\u6bcf\u4e2a\u6587\u4ef6\u4e2d\u7684\u56fe\u50cf\u548c\u6807\u6ce8\u4fe1\u606f\u3002\u5bf9\u4e8e\u6bcf\u4e2a\u56fe\u50cf&#xff0c;\u51fd\u6570\u5c06\u6807\u6ce8\u4fe1\u606f\u8f6c\u6362\u4e3aYOLO\u683c\u5f0f&#xff0c;\u5305\u62ec\u5f52\u4e00\u5316\u8fb9\u754c\u6846\u7684\u5750\u6807\u3001\u7c7b\u522bID\u7b49&#xff0c;\u5e76\u5c06\u7ed3\u679c\u5199\u5165\u6587\u672c\u6587\u4ef6\u4e2d\u3002<\/p>\n<p>\u5728\u5904\u7406\u6807\u6ce8\u65f6&#xff0c;\u51fd\u6570\u4f1a\u68c0\u67e5\u6bcf\u4e2a\u6807\u6ce8\u662f\u5426\u4e3a\u201ciscrowd\u201d&#xff0c;\u5982\u679c\u662f&#xff0c;\u5219\u8df3\u8fc7\u8be5\u6807\u6ce8\u3002\u5bf9\u4e8e\u6709\u6548\u7684\u6807\u6ce8&#xff0c;\u51fd\u6570\u4f1a\u5c06\u8fb9\u754c\u6846\u7684\u5750\u6807\u4eceCOCO\u683c\u5f0f&#xff08;\u5de6\u4e0a\u89d2\u5750\u6807\u548c\u5bbd\u9ad8&#xff09;\u8f6c\u6362\u4e3aYOLO\u683c\u5f0f&#xff08;\u4e2d\u5fc3\u5750\u6807\u548c\u5bbd\u9ad8&#xff09;&#xff0c;\u5e76\u8fdb\u884c\u5f52\u4e00\u5316\u5904\u7406\u3002\u82e5\u9009\u62e9\u4f7f\u7528\u5206\u5272\u63a9\u7801\u548c\u5173\u952e\u70b9&#xff0c;\u51fd\u6570\u4e5f\u4f1a\u5904\u7406\u8fd9\u4e9b\u4fe1\u606f\u5e76\u5c06\u5176\u5199\u5165\u6587\u4ef6\u3002<\/p>\n<p>convert_dota_to_yolo_obb\u51fd\u6570\u5219\u662f\u7528\u4e8e\u5c06DOTA\u6570\u636e\u96c6\u7684\u6807\u6ce8\u8f6c\u6362\u4e3aYOLO\u7684\u6709\u5411\u8fb9\u754c\u6846&#xff08;OBB&#xff09;\u683c\u5f0f\u3002\u5b83\u4f1a\u8bfb\u53d6DOTA\u6570\u636e\u96c6\u4e2d\u8bad\u7ec3\u548c\u9a8c\u8bc1\u96c6\u7684\u56fe\u50cf\u548c\u539f\u59cb\u6807\u6ce8&#xff0c;\u5e76\u5c06\u8f6c\u6362\u540e\u7684\u6807\u6ce8\u4fdd\u5b58\u5230\u65b0\u7684\u76ee\u5f55\u4e2d\u3002\u51fd\u6570\u4e2d\u5b9a\u4e49\u4e86\u4e00\u4e2aconvert_label\u7684\u5185\u90e8\u51fd\u6570&#xff0c;\u7528\u4e8e\u5904\u7406\u5355\u4e2a\u56fe\u50cf\u7684\u6807\u6ce8\u8f6c\u6362&#xff0c;\u786e\u4fdd\u5750\u6807\u5f52\u4e00\u5316\u5e76\u683c\u5f0f\u5316\u4e3aYOLO\u6240\u9700\u7684\u8f93\u51fa\u683c\u5f0f\u3002<\/p>\n<p>\u6b64\u5916&#xff0c;min_index\u51fd\u6570\u7528\u4e8e\u67e5\u627e\u4e24\u4e2a\u4e8c\u7ef4\u70b9\u6570\u7ec4\u4e4b\u95f4\u8ddd\u79bb\u6700\u77ed\u7684\u70b9\u5bf9\u7d22\u5f15&#xff0c;\u800cmerge_multi_segment\u51fd\u6570\u5219\u7528\u4e8e\u5408\u5e76\u591a\u4e2a\u5206\u5272\u533a\u57df&#xff0c;\u5c06\u5b83\u4eec\u8fde\u63a5\u6210\u4e00\u4e2a\u5b8c\u6574\u7684\u533a\u57df\u3002\u8fd9\u5728\u5904\u7406\u590d\u6742\u7684\u5206\u5272\u6807\u6ce8\u65f6\u975e\u5e38\u6709\u7528\u3002<\/p>\n<p>\u603b\u4f53\u800c\u8a00&#xff0c;\u8fd9\u4e2a\u7a0b\u5e8f\u6587\u4ef6\u4e3aYOLO\u6a21\u578b\u63d0\u4f9b\u4e86\u4eceCOCO\u548cDOTA\u6570\u636e\u96c6\u4e2d\u8bfb\u53d6\u548c\u8f6c\u6362\u6807\u6ce8\u7684\u529f\u80fd&#xff0c;\u4fbf\u4e8e\u7528\u6237\u8fdb\u884c\u76ee\u6807\u68c0\u6d4b\u6a21\u578b\u7684\u8bad\u7ec3\u3002<\/p>\n<p><span class=\"token comment\"># \u5bfc\u5165\u6240\u9700\u7684\u6a21\u578b\u548c\u51fd\u6570<\/span><br \/>\n<span class=\"token keyword\">from<\/span> <span class=\"token punctuation\">.<\/span>tasks <span class=\"token keyword\">import<\/span> <span class=\"token punctuation\">(<\/span>BaseModel<span class=\"token punctuation\">,<\/span> ClassificationModel<span class=\"token punctuation\">,<\/span> DetectionModel<span class=\"token punctuation\">,<\/span> SegmentationModel<span class=\"token punctuation\">,<\/span><br \/>\n                    attempt_load_one_weight<span class=\"token punctuation\">,<\/span> attempt_load_weights<span class=\"token punctuation\">,<\/span> guess_model_scale<span class=\"token punctuation\">,<\/span><br \/>\n                    guess_model_task<span class=\"token punctuation\">,<\/span> parse_model<span class=\"token punctuation\">,<\/span> torch_safe_load<span class=\"token punctuation\">,<\/span> yaml_model_load<span class=\"token punctuation\">)<\/span><\/p>\n<p><span class=\"token comment\"># \u5b9a\u4e49\u6a21\u5757\u7684\u516c\u5f00\u63a5\u53e3&#xff0c;\u5305\u542b\u53ef\u4f9b\u5916\u90e8\u4f7f\u7528\u7684\u7c7b\u548c\u51fd\u6570<\/span><br \/>\n__all__ <span class=\"token operator\">&#061;<\/span> <span class=\"token punctuation\">(<\/span><span class=\"token string\">&#039;attempt_load_one_weight&#039;<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token string\">&#039;attempt_load_weights&#039;<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token string\">&#039;parse_model&#039;<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token string\">&#039;yaml_model_load&#039;<\/span><span class=\"token punctuation\">,<\/span><br \/>\n           <span class=\"token string\">&#039;guess_model_task&#039;<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token string\">&#039;guess_model_scale&#039;<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token string\">&#039;torch_safe_load&#039;<\/span><span class=\"token punctuation\">,<\/span><br \/>\n           <span class=\"token string\">&#039;DetectionModel&#039;<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token string\">&#039;SegmentationModel&#039;<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token string\">&#039;ClassificationModel&#039;<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token string\">&#039;BaseModel&#039;<\/span><span class=\"token punctuation\">)<\/span><\/p>\n<h4>\u4ee3\u7801\u6838\u5fc3\u90e8\u5206\u53ca\u6ce8\u91ca&#xff1a;<\/h4>\n<li>\n<p>\u5bfc\u5165\u6a21\u5757&#xff1a;<\/p>\n<ul>\n<li>from .tasks import &#8230;&#xff1a;\u4ece\u5f53\u524d\u5305\u7684 tasks \u6a21\u5757\u4e2d\u5bfc\u5165\u591a\u4e2a\u7c7b\u548c\u51fd\u6570\u3002\u8fd9\u4e9b\u7c7b\u548c\u51fd\u6570\u662fYOLO\u6a21\u578b\u5b9e\u73b0\u7684\u6838\u5fc3\u90e8\u5206&#xff0c;\u5206\u522b\u7528\u4e8e\u4e0d\u540c\u7684\u4efb\u52a1&#xff08;\u5982\u5206\u7c7b\u3001\u68c0\u6d4b\u3001\u5206\u5272\u7b49&#xff09;\u3002<\/li>\n<\/ul>\n<\/li>\n<li>\n<p>\u516c\u5f00\u63a5\u53e3\u5b9a\u4e49&#xff1a;<\/p>\n<ul>\n<li>__all__ &#061; (&#8230;)&#xff1a;\u8fd9\u662f\u4e00\u4e2a\u7279\u6b8a\u7684\u53d8\u91cf&#xff0c;\u7528\u4e8e\u5b9a\u4e49\u5f53\u4f7f\u7528 from module import * \u65f6&#xff0c;\u54ea\u4e9b\u7c7b\u548c\u51fd\u6570\u662f\u53ef\u4ee5\u88ab\u5bfc\u51fa\u7684\u3002\u8fd9\u91cc\u5217\u51fa\u4e86\u6240\u6709\u9700\u8981\u516c\u5f00\u7684\u6a21\u578b\u7c7b\u548c\u8f85\u52a9\u51fd\u6570&#xff0c;\u786e\u4fdd\u7528\u6237\u53ef\u4ee5\u65b9\u4fbf\u5730\u8bbf\u95ee\u8fd9\u4e9b\u529f\u80fd\u3002<\/li>\n<\/ul>\n<\/li>\n<h4>\u4e3b\u8981\u529f\u80fd\u6982\u8ff0&#xff1a;<\/h4>\n<ul>\n<li>\n<p>\u6a21\u578b\u7c7b&#xff1a;<\/p>\n<ul>\n<li>BaseModel&#xff1a;\u57fa\u7840\u6a21\u578b\u7c7b&#xff0c;\u53ef\u80fd\u5305\u542b\u6240\u6709\u6a21\u578b\u5171\u4eab\u7684\u57fa\u672c\u529f\u80fd\u3002<\/li>\n<li>ClassificationModel&#xff1a;\u7528\u4e8e\u56fe\u50cf\u5206\u7c7b\u7684\u6a21\u578b\u3002<\/li>\n<li>DetectionModel&#xff1a;\u7528\u4e8e\u76ee\u6807\u68c0\u6d4b\u7684\u6a21\u578b\u3002<\/li>\n<li>SegmentationModel&#xff1a;\u7528\u4e8e\u56fe\u50cf\u5206\u5272\u7684\u6a21\u578b\u3002<\/li>\n<\/ul>\n<\/li>\n<li>\n<p>\u8f85\u52a9\u51fd\u6570&#xff1a;<\/p>\n<ul>\n<li>attempt_load_one_weight&#xff1a;\u5c1d\u8bd5\u52a0\u8f7d\u5355\u4e2a\u6743\u91cd\u6587\u4ef6\u3002<\/li>\n<li>attempt_load_weights&#xff1a;\u5c1d\u8bd5\u52a0\u8f7d\u591a\u4e2a\u6743\u91cd\u6587\u4ef6\u3002<\/li>\n<li>guess_model_scale&#xff1a;\u6839\u636e\u8f93\u5165\u63a8\u6d4b\u6a21\u578b\u7684\u89c4\u6a21\u3002<\/li>\n<li>guess_model_task&#xff1a;\u6839\u636e\u8f93\u5165\u63a8\u6d4b\u6a21\u578b\u7684\u4efb\u52a1\u7c7b\u578b\u3002<\/li>\n<li>parse_model&#xff1a;\u89e3\u6790\u6a21\u578b\u7ed3\u6784\u3002<\/li>\n<li>torch_safe_load&#xff1a;\u5b89\u5168\u52a0\u8f7dPyTorch\u6a21\u578b\u3002<\/li>\n<li>yaml_model_load&#xff1a;\u4eceYAML\u6587\u4ef6\u52a0\u8f7d\u6a21\u578b\u914d\u7f6e\u3002&#096;&#096;&#096;<br \/>\n\u8fd9\u4e2a\u7a0b\u5e8f\u6587\u4ef6\u662fUltralytics YOLO\u9879\u76ee\u7684\u4e00\u90e8\u5206&#xff0c;\u4e3b\u8981\u7528\u4e8e\u5bfc\u5165\u548c\u7ba1\u7406\u4e0e\u6a21\u578b\u76f8\u5173\u7684\u529f\u80fd\u548c\u7c7b\u3002\u6587\u4ef6\u5f00\u5934\u7684\u6ce8\u91ca\u8868\u660e\u8be5\u9879\u76ee\u9075\u5faaAGPL-3.0\u8bb8\u53ef\u8bc1&#xff0c;\u610f\u5473\u7740\u5b83\u662f\u4e00\u4e2a\u5f00\u6e90\u9879\u76ee\u3002<\/li>\n<\/ul>\n<\/li>\n<\/ul>\n<p>\u5728\u4ee3\u7801\u4e2d&#xff0c;\u9996\u5148\u4ece\u540c\u4e00\u5305&#xff08;\u5373\u5f53\u524d\u76ee\u5f55&#xff09;\u4e2d\u5bfc\u5165\u4e86\u4e00\u4e9b\u6a21\u5757\u548c\u7c7b&#xff0c;\u5305\u62ec\u57fa\u7840\u6a21\u578b\u7c7b&#xff08;BaseModel&#xff09;\u3001\u5206\u7c7b\u6a21\u578b\u7c7b&#xff08;ClassificationModel&#xff09;\u3001\u68c0\u6d4b\u6a21\u578b\u7c7b&#xff08;DetectionModel&#xff09;\u3001\u5206\u5272\u6a21\u578b\u7c7b&#xff08;SegmentationModel&#xff09;\u4ee5\u53ca\u4e00\u4e9b\u4e0e\u6a21\u578b\u52a0\u8f7d\u548c\u89e3\u6790\u76f8\u5173\u7684\u51fd\u6570\u3002\u8fd9\u4e9b\u51fd\u6570\u5305\u62ec\u5c1d\u8bd5\u52a0\u8f7d\u5355\u4e2a\u6743\u91cd&#xff08;attempt_load_one_weight&#xff09;\u3001\u5c1d\u8bd5\u52a0\u8f7d\u6743\u91cd&#xff08;attempt_load_weights&#xff09;\u3001\u731c\u6d4b\u6a21\u578b\u7684\u7f29\u653e&#xff08;guess_model_scale&#xff09;\u3001\u731c\u6d4b\u6a21\u578b\u7684\u4efb\u52a1&#xff08;guess_model_task&#xff09;\u3001\u89e3\u6790\u6a21\u578b&#xff08;parse_model&#xff09;\u3001\u5b89\u5168\u52a0\u8f7dTorch\u6a21\u578b&#xff08;torch_safe_load&#xff09;\u548c\u4eceYAML\u6587\u4ef6\u52a0\u8f7d\u6a21\u578b&#xff08;yaml_model_load&#xff09;\u3002<\/p>\n<p>\u6700\u540e&#xff0c;\u4f7f\u7528__all__\u53d8\u91cf\u5b9a\u4e49\u4e86\u4e00\u4e2a\u516c\u5171\u63a5\u53e3&#xff0c;\u5217\u51fa\u4e86\u53ef\u4ee5\u88ab\u5916\u90e8\u5bfc\u5165\u7684\u6a21\u5757\u548c\u7c7b\u3002\u8fd9\u610f\u5473\u7740&#xff0c;\u5f53\u5176\u4ed6\u6a21\u5757\u4f7f\u7528from ultralytics.nn import *\u8fd9\u6837\u7684\u8bed\u53e5\u65f6&#xff0c;\u53ea\u4f1a\u5bfc\u5165__all__\u4e2d\u5217\u51fa\u7684\u5185\u5bb9\u3002\u8fd9\u79cd\u505a\u6cd5\u6709\u52a9\u4e8e\u63a7\u5236\u6a21\u5757\u7684\u53ef\u89c1\u6027&#xff0c;\u786e\u4fdd\u53ea\u66b4\u9732\u5fc5\u8981\u7684\u529f\u80fd\u7ed9\u7528\u6237&#xff0c;\u4ece\u800c\u63d0\u9ad8\u4ee3\u7801\u7684\u53ef\u7ef4\u62a4\u6027\u548c\u53ef\u8bfb\u6027\u3002<\/p>\n<p>&#096;&#096;&#096;python<br \/>\nimport os<br \/>\nimport torch<br \/>\nimport yaml<br \/>\nfrom ultralytics import YOLO  # \u5bfc\u5165YOLO\u6a21\u578b<\/p>\n<p>if __name__ &#061;&#061; &#039;__main__&#039;:  # \u786e\u4fdd\u8be5\u6a21\u5757\u88ab\u76f4\u63a5\u8fd0\u884c\u65f6\u624d\u6267\u884c\u4ee5\u4e0b\u4ee3\u7801<br \/>\n    # \u8bbe\u7f6e\u8bad\u7ec3\u53c2\u6570<br \/>\n    workers &#061; 1  # \u6570\u636e\u52a0\u8f7d\u7684\u5de5\u4f5c\u8fdb\u7a0b\u6570<br \/>\n    batch &#061; 8  # \u6bcf\u4e2a\u6279\u6b21\u7684\u6837\u672c\u6570\u91cf<br \/>\n    device &#061; &#034;0&#034; if torch.cuda.is_available() else &#034;cpu&#034;  # \u68c0\u67e5\u662f\u5426\u6709\u53ef\u7528\u7684GPU&#xff0c;\u5982\u679c\u6ca1\u6709\u5219\u4f7f\u7528CPU<\/p>\n<p>    # \u83b7\u53d6\u6570\u636e\u96c6\u7684YAML\u914d\u7f6e\u6587\u4ef6\u7684\u7edd\u5bf9\u8def\u5f84<br \/>\n    data_path &#061; abs_path(f&#039;datasets\/data\/data.yaml&#039;, path_type&#061;&#039;current&#039;)<\/p>\n<p>    # \u5c06\u8def\u5f84\u683c\u5f0f\u8f6c\u6362\u4e3aUnix\u98ce\u683c<br \/>\n    unix_style_path &#061; data_path.replace(os.sep, &#039;\/&#039;)<br \/>\n    # \u83b7\u53d6\u6570\u636e\u96c6\u76ee\u5f55\u8def\u5f84<br \/>\n    directory_path &#061; os.path.dirname(unix_style_path)<\/p>\n<p>    # \u8bfb\u53d6YAML\u6587\u4ef6&#xff0c;\u4fdd\u6301\u539f\u6709\u987a\u5e8f<br \/>\n    with open(data_path, &#039;r&#039;) as file:<br \/>\n        data &#061; yaml.load(file, Loader&#061;yaml.FullLoader)<\/p>\n<p>    # \u4fee\u6539\u6570\u636e\u96c6\u8def\u5f84<br \/>\n    if &#039;train&#039; in data and &#039;val&#039; in data and &#039;test&#039; in data:<br \/>\n        data[&#039;train&#039;] &#061; directory_path &#043; &#039;\/train&#039;  # \u66f4\u65b0\u8bad\u7ec3\u96c6\u8def\u5f84<br \/>\n        data[&#039;val&#039;] &#061; directory_path &#043; &#039;\/val&#039;      # \u66f4\u65b0\u9a8c\u8bc1\u96c6\u8def\u5f84<br \/>\n        data[&#039;test&#039;] &#061; directory_path &#043; &#039;\/test&#039;    # \u66f4\u65b0\u6d4b\u8bd5\u96c6\u8def\u5f84<\/p>\n<p>        # \u5c06\u4fee\u6539\u540e\u7684\u6570\u636e\u5199\u56deYAML\u6587\u4ef6<br \/>\n        with open(data_path, &#039;w&#039;) as file:<br \/>\n            yaml.safe_dump(data, file, sort_keys&#061;False)<\/p>\n<p>    # \u52a0\u8f7dYOLO\u6a21\u578b\u914d\u7f6e\u548c\u9884\u8bad\u7ec3\u6743\u91cd<br \/>\n    model &#061; YOLO(r&#034;C:\\\\codeseg\\\\codenew\\\\50&#043;\u79cdYOLOv8\u7b97\u6cd5\u6539\u8fdb\u6e90\u7801\u5927\u5168\u548c\u8c03\u8bd5\u52a0\u8f7d\u8bad\u7ec3\u6559\u7a0b&#xff08;\u975e\u5fc5\u8981&#xff09;\\\\\u6539\u8fdbYOLOv8\u6a21\u578b\u914d\u7f6e\u6587\u4ef6\\\\yolov8-seg-C2f-Faster.yaml&#034;).load(&#034;.\/weights\/yolov8s-seg.pt&#034;)<\/p>\n<p>    # \u5f00\u59cb\u8bad\u7ec3\u6a21\u578b<br \/>\n    results &#061; model.train(<br \/>\n        data&#061;data_path,  # \u6307\u5b9a\u8bad\u7ec3\u6570\u636e\u7684\u914d\u7f6e\u6587\u4ef6\u8def\u5f84<br \/>\n        device&#061;device,  # \u6307\u5b9a\u4f7f\u7528\u7684\u8bbe\u5907&#xff08;GPU\u6216CPU&#xff09;<br \/>\n        workers&#061;workers,  # \u6307\u5b9a\u4f7f\u7528\u7684\u5de5\u4f5c\u8fdb\u7a0b\u6570<br \/>\n        imgsz&#061;640,  # \u6307\u5b9a\u8f93\u5165\u56fe\u50cf\u7684\u5927\u5c0f\u4e3a640&#215;640<br \/>\n        epochs&#061;100,  # \u6307\u5b9a\u8bad\u7ec3\u7684\u8f6e\u6570\u4e3a100<br \/>\n        batch&#061;batch,  # \u6307\u5b9a\u6bcf\u4e2a\u6279\u6b21\u7684\u6837\u672c\u6570\u91cf<br \/>\n    )<\/p>\n<h4>\u4ee3\u7801\u6838\u5fc3\u90e8\u5206\u8bf4\u660e&#xff1a;<\/h4>\n<li>\u5bfc\u5165\u5fc5\u8981\u7684\u5e93&#xff1a;\u5bfc\u5165\u4e86\u5904\u7406\u6587\u4ef6\u8def\u5f84\u3001\u6df1\u5ea6\u5b66\u4e60\u6846\u67b6&#xff08;PyTorch&#xff09;\u3001YAML\u6587\u4ef6\u5904\u7406\u548cYOLO\u6a21\u578b\u7684\u5e93\u3002<\/li>\n<li>\u53c2\u6570\u8bbe\u7f6e&#xff1a;\u5b9a\u4e49\u4e86\u8bad\u7ec3\u8fc7\u7a0b\u4e2d\u4f7f\u7528\u7684\u53c2\u6570&#xff0c;\u5305\u62ec\u5de5\u4f5c\u8fdb\u7a0b\u6570\u3001\u6279\u6b21\u5927\u5c0f\u548c\u8bbe\u5907\u9009\u62e9&#xff08;GPU\u6216CPU&#xff09;\u3002<\/li>\n<li>\u6570\u636e\u96c6\u8def\u5f84\u5904\u7406&#xff1a;\u8bfb\u53d6YAML\u914d\u7f6e\u6587\u4ef6&#xff0c;\u66f4\u65b0\u8bad\u7ec3\u3001\u9a8c\u8bc1\u548c\u6d4b\u8bd5\u6570\u636e\u96c6\u7684\u8def\u5f84&#xff0c;\u5e76\u5c06\u5176\u5199\u56de\u6587\u4ef6\u3002<\/li>\n<li>\u6a21\u578b\u52a0\u8f7d&#xff1a;\u52a0\u8f7dYOLO\u6a21\u578b\u7684\u914d\u7f6e\u6587\u4ef6\u548c\u9884\u8bad\u7ec3\u6743\u91cd\u3002<\/li>\n<li>\u6a21\u578b\u8bad\u7ec3&#xff1a;\u8c03\u7528train\u65b9\u6cd5\u5f00\u59cb\u8bad\u7ec3\u6a21\u578b&#xff0c;\u4f20\u5165\u5fc5\u8981\u7684\u53c2\u6570\u3002&#096;&#096;&#096;<br \/>\n\u8be5\u7a0b\u5e8f\u6587\u4ef6 train.py \u662f\u4e00\u4e2a\u7528\u4e8e\u8bad\u7ec3 YOLO&#xff08;You Only Look Once&#xff09;\u6a21\u578b\u7684\u811a\u672c\u3002\u9996\u5148&#xff0c;\u7a0b\u5e8f\u5bfc\u5165\u4e86\u5fc5\u8981\u7684\u5e93&#xff0c;\u5305\u62ec os\u3001torch\u3001yaml \u548c ultralytics \u4e2d\u7684 YOLO \u6a21\u578b\u3002\u6b64\u5916&#xff0c;\u8fd8\u5bfc\u5165\u4e86 QtFusion.path \u4e2d\u7684 abs_path \u51fd\u6570\u7528\u4e8e\u5904\u7406\u8def\u5f84&#xff0c;\u4ee5\u53ca matplotlib \u7528\u4e8e\u56fe\u5f62\u663e\u793a\u3002<\/li>\n<p>\u5728 __main__ \u5757\u4e2d&#xff0c;\u7a0b\u5e8f\u9996\u5148\u8bbe\u7f6e\u4e86\u4e00\u4e9b\u8bad\u7ec3\u53c2\u6570&#xff0c;\u5305\u62ec\u5de5\u4f5c\u8fdb\u7a0b\u6570 workers\u3001\u6279\u6b21\u5927\u5c0f batch \u548c\u8bbe\u5907\u7c7b\u578b device\u3002\u8bbe\u5907\u7c7b\u578b\u6839\u636e\u662f\u5426\u6709\u53ef\u7528\u7684 GPU \u6765\u51b3\u5b9a&#xff0c;\u5982\u679c\u6709\u5219\u4f7f\u7528 GPU&#xff08;\u201c0\u201d&#xff09;&#xff0c;\u5426\u5219\u4f7f\u7528 CPU&#xff08;\u201ccpu\u201d&#xff09;\u3002\u63a5\u7740&#xff0c;\u7a0b\u5e8f\u901a\u8fc7 abs_path \u51fd\u6570\u83b7\u53d6\u6570\u636e\u96c6\u914d\u7f6e\u6587\u4ef6\u7684\u7edd\u5bf9\u8def\u5f84&#xff0c;\u8fd9\u4e2a\u914d\u7f6e\u6587\u4ef6\u662f\u4e00\u4e2a YAML \u683c\u5f0f\u7684\u6587\u4ef6&#xff0c;\u5305\u542b\u4e86\u8bad\u7ec3\u3001\u9a8c\u8bc1\u548c\u6d4b\u8bd5\u6570\u636e\u7684\u8def\u5f84\u3002<\/p>\n<p>\u7a0b\u5e8f\u5c06\u83b7\u53d6\u5230\u7684\u8def\u5f84\u8f6c\u6362\u4e3a UNIX \u98ce\u683c\u7684\u8def\u5f84&#xff0c;\u5e76\u63d0\u53d6\u51fa\u76ee\u5f55\u8def\u5f84\u3002\u7136\u540e&#xff0c;\u7a0b\u5e8f\u6253\u5f00 YAML \u6587\u4ef6\u5e76\u8bfb\u53d6\u5176\u5185\u5bb9\u3002\u82e5 YAML \u6587\u4ef6\u4e2d\u5305\u542b train\u3001val \u548c test \u9879&#xff0c;\u7a0b\u5e8f\u4f1a\u5c06\u8fd9\u4e9b\u9879\u7684\u8def\u5f84\u4fee\u6539\u4e3a\u76f8\u5bf9\u4e8e\u76ee\u5f55\u8def\u5f84\u7684\u5177\u4f53\u8def\u5f84&#xff0c;\u5e76\u5c06\u4fee\u6539\u540e\u7684\u5185\u5bb9\u5199\u56de\u5230 YAML \u6587\u4ef6\u4e2d\u3002<\/p>\n<p>\u63a5\u4e0b\u6765&#xff0c;\u7a0b\u5e8f\u52a0\u8f7d\u4e86\u4e00\u4e2a YOLO \u6a21\u578b\u7684\u914d\u7f6e\u6587\u4ef6&#xff0c;\u5e76\u901a\u8fc7\u6307\u5b9a\u7684\u6743\u91cd\u6587\u4ef6\u52a0\u8f7d\u9884\u8bad\u7ec3\u6a21\u578b\u3002\u6a21\u578b\u7684\u914d\u7f6e\u6587\u4ef6\u8def\u5f84\u548c\u6743\u91cd\u6587\u4ef6\u8def\u5f84\u662f\u786c\u7f16\u7801\u7684&#xff0c;\u7528\u6237\u53ef\u4ee5\u6839\u636e\u9700\u8981\u8fdb\u884c\u4fee\u6539\u3002<\/p>\n<p>\u6700\u540e&#xff0c;\u7a0b\u5e8f\u8c03\u7528 model.train \u65b9\u6cd5\u5f00\u59cb\u8bad\u7ec3\u6a21\u578b&#xff0c;\u4f20\u5165\u7684\u6570\u636e\u914d\u7f6e\u6587\u4ef6\u8def\u5f84\u3001\u8bbe\u5907\u7c7b\u578b\u3001\u5de5\u4f5c\u8fdb\u7a0b\u6570\u3001\u8f93\u5165\u56fe\u50cf\u5927\u5c0f&#xff08;640&#215;640&#xff09;\u3001\u8bad\u7ec3\u7684 epoch \u6570&#xff08;100&#xff09;\u548c\u6279\u6b21\u5927\u5c0f&#xff08;8&#xff09;\u7b49\u53c2\u6570\u3002\u901a\u8fc7\u8fd9\u4e9b\u8bbe\u7f6e&#xff0c;\u7a0b\u5e8f\u5c06\u5f00\u59cb\u5bf9\u6307\u5b9a\u7684\u6570\u636e\u96c6\u8fdb\u884c\u8bad\u7ec3&#xff0c;\u6700\u7ec8\u8f93\u51fa\u8bad\u7ec3\u7ed3\u679c\u3002<\/p>\n<p>&#096;&#096;&#096;python<br \/>\n# \u5bfc\u5165\u5fc5\u8981\u7684\u5e93\u548c\u6a21\u5757<br \/>\nfrom copy import copy<br \/>\nfrom ultralytics.models import yolo<br \/>\nfrom ultralytics.nn.tasks import PoseModel<br \/>\nfrom ultralytics.utils import DEFAULT_CFG, LOGGER<br \/>\nfrom ultralytics.utils.plotting import plot_images, plot_results<\/p>\n<p>class PoseTrainer(yolo.detect.DetectionTrainer):<br \/>\n    &#034;&#034;&#034;<br \/>\n    PoseTrainer\u7c7b&#xff0c;\u6269\u5c55\u4e86DetectionTrainer\u7c7b&#xff0c;\u7528\u4e8e\u57fa\u4e8e\u59ff\u6001\u6a21\u578b\u7684\u8bad\u7ec3\u3002<br \/>\n    &#034;&#034;&#034;<\/p>\n<p>    def __init__(self, cfg&#061;DEFAULT_CFG, overrides&#061;None, _callbacks&#061;None):<br \/>\n        &#034;&#034;&#034;\u521d\u59cb\u5316PoseTrainer\u5bf9\u8c61&#xff0c;\u6307\u5b9a\u914d\u7f6e\u548c\u8986\u76d6\u53c2\u6570\u3002&#034;&#034;&#034;<br \/>\n        if overrides is None:<br \/>\n            overrides &#061; {}<br \/>\n        overrides[&#039;task&#039;] &#061; &#039;pose&#039;  # \u8bbe\u7f6e\u4efb\u52a1\u7c7b\u578b\u4e3a\u59ff\u6001\u4f30\u8ba1<br \/>\n        super().__init__(cfg, overrides, _callbacks)  # \u8c03\u7528\u7236\u7c7b\u6784\u9020\u51fd\u6570<\/p>\n<p>        # \u9488\u5bf9Apple MPS\u8bbe\u5907\u7684\u8b66\u544a<br \/>\n        if isinstance(self.args.device, str) and self.args.device.lower() &#061;&#061; &#039;mps&#039;:<br \/>\n            LOGGER.warning(&#034;WARNING \u26a0\ufe0f Apple MPS known Pose bug. Recommend &#039;device&#061;cpu&#039; for Pose models.&#034;)<\/p>\n<p>    def get_model(self, cfg&#061;None, weights&#061;None, verbose&#061;True):<br \/>\n        &#034;&#034;&#034;\u83b7\u53d6\u6307\u5b9a\u914d\u7f6e\u548c\u6743\u91cd\u7684\u59ff\u6001\u4f30\u8ba1\u6a21\u578b\u3002&#034;&#034;&#034;<br \/>\n        # \u521b\u5efaPoseModel\u5b9e\u4f8b<br \/>\n        model &#061; PoseModel(cfg, ch&#061;3, nc&#061;self.data[&#039;nc&#039;], data_kpt_shape&#061;self.data[&#039;kpt_shape&#039;], verbose&#061;verbose)<br \/>\n        if weights:<br \/>\n            model.load(weights)  # \u52a0\u8f7d\u6743\u91cd<\/p>\n<p>        return model  # \u8fd4\u56de\u6a21\u578b<\/p>\n<p>    def set_model_attributes(self):<br \/>\n        &#034;&#034;&#034;\u8bbe\u7f6ePoseModel\u7684\u5173\u952e\u70b9\u5f62\u72b6\u5c5e\u6027\u3002&#034;&#034;&#034;<br \/>\n        super().set_model_attributes()  # \u8c03\u7528\u7236\u7c7b\u65b9\u6cd5<br \/>\n        self.model.kpt_shape &#061; self.data[&#039;kpt_shape&#039;]  # \u8bbe\u7f6e\u5173\u952e\u70b9\u5f62\u72b6<\/p>\n<p>    def get_validator(self):<br \/>\n        &#034;&#034;&#034;\u8fd4\u56dePoseValidator\u7c7b\u7684\u5b9e\u4f8b&#xff0c;\u7528\u4e8e\u9a8c\u8bc1\u3002&#034;&#034;&#034;<br \/>\n        self.loss_names &#061; &#039;box_loss&#039;, &#039;pose_loss&#039;, &#039;kobj_loss&#039;, &#039;cls_loss&#039;, &#039;dfl_loss&#039;  # \u5b9a\u4e49\u635f\u5931\u540d\u79f0<br \/>\n        return yolo.pose.PoseValidator(self.test_loader, save_dir&#061;self.save_dir, args&#061;copy(self.args))  # \u8fd4\u56de\u9a8c\u8bc1\u5668\u5b9e\u4f8b<\/p>\n<p>    def plot_training_samples(self, batch, ni):<br \/>\n        &#034;&#034;&#034;\u7ed8\u5236\u4e00\u6279\u8bad\u7ec3\u6837\u672c&#xff0c;\u5305\u62ec\u7c7b\u522b\u6807\u7b7e\u3001\u8fb9\u754c\u6846\u548c\u5173\u952e\u70b9\u3002&#034;&#034;&#034;<br \/>\n        images &#061; batch[&#039;img&#039;]  # \u83b7\u53d6\u56fe\u50cf<br \/>\n        kpts &#061; batch[&#039;keypoints&#039;]  # \u83b7\u53d6\u5173\u952e\u70b9<br \/>\n        cls &#061; batch[&#039;cls&#039;].squeeze(-1)  # \u83b7\u53d6\u7c7b\u522b<br \/>\n        bboxes &#061; batch[&#039;bboxes&#039;]  # \u83b7\u53d6\u8fb9\u754c\u6846<br \/>\n        paths &#061; batch[&#039;im_file&#039;]  # \u83b7\u53d6\u56fe\u50cf\u6587\u4ef6\u8def\u5f84<br \/>\n        batch_idx &#061; batch[&#039;batch_idx&#039;]  # \u83b7\u53d6\u6279\u6b21\u7d22\u5f15<\/p>\n<p>        # \u7ed8\u5236\u56fe\u50cf<br \/>\n        plot_images(images,<br \/>\n                    batch_idx,<br \/>\n                    cls,<br \/>\n                    bboxes,<br \/>\n                    kpts&#061;kpts,<br \/>\n                    paths&#061;paths,<br \/>\n                    fname&#061;self.save_dir \/ f&#039;train_batch{ni}.jpg&#039;,  # \u4fdd\u5b58\u56fe\u50cf\u6587\u4ef6<br \/>\n                    on_plot&#061;self.on_plot)<\/p>\n<p>    def plot_metrics(self):<br \/>\n        &#034;&#034;&#034;\u7ed8\u5236\u8bad\u7ec3\u548c\u9a8c\u8bc1\u6307\u6807\u3002&#034;&#034;&#034;<br \/>\n        plot_results(file&#061;self.csv, pose&#061;True, on_plot&#061;self.on_plot)  # \u4fdd\u5b58\u7ed3\u679c\u56fe\u50cf<\/p>\n<h4>\u4ee3\u7801\u8bf4\u660e&#xff1a;<\/h4>\n<li>PoseTrainer\u7c7b&#xff1a;\u8fd9\u662f\u4e00\u4e2a\u7528\u4e8e\u59ff\u6001\u4f30\u8ba1\u8bad\u7ec3\u7684\u7c7b&#xff0c;\u7ee7\u627f\u81eaYOLO\u7684\u68c0\u6d4b\u8bad\u7ec3\u5668\u3002<\/li>\n<li>\u521d\u59cb\u5316\u65b9\u6cd5&#xff1a;\u8bbe\u7f6e\u4efb\u52a1\u7c7b\u578b\u4e3a\u59ff\u6001\u4f30\u8ba1&#xff0c;\u5e76\u5904\u7406\u8bbe\u5907\u8b66\u544a\u3002<\/li>\n<li>\u83b7\u53d6\u6a21\u578b&#xff1a;\u6839\u636e\u914d\u7f6e\u548c\u6743\u91cd\u521b\u5efa\u5e76\u8fd4\u56de\u59ff\u6001\u6a21\u578b\u3002<\/li>\n<li>\u8bbe\u7f6e\u6a21\u578b\u5c5e\u6027&#xff1a;\u8bbe\u7f6e\u6a21\u578b\u7684\u5173\u952e\u70b9\u5f62\u72b6\u3002<\/li>\n<li>\u83b7\u53d6\u9a8c\u8bc1\u5668&#xff1a;\u8fd4\u56de\u7528\u4e8e\u9a8c\u8bc1\u7684PoseValidator\u5b9e\u4f8b&#xff0c;\u5e76\u5b9a\u4e49\u635f\u5931\u540d\u79f0\u3002<\/li>\n<li>\u7ed8\u5236\u8bad\u7ec3\u6837\u672c&#xff1a;\u7ed8\u5236\u8bad\u7ec3\u8fc7\u7a0b\u4e2d\u4e00\u6279\u6837\u672c\u7684\u56fe\u50cf&#xff0c;\u5305\u62ec\u7c7b\u522b\u3001\u8fb9\u754c\u6846\u548c\u5173\u952e\u70b9\u3002<\/li>\n<li>\u7ed8\u5236\u6307\u6807&#xff1a;\u7ed8\u5236\u8bad\u7ec3\u548c\u9a8c\u8bc1\u7684\u6027\u80fd\u6307\u6807\u56fe\u3002&#096;&#096;&#096;<br \/>\n\u8fd9\u4e2a\u7a0b\u5e8f\u6587\u4ef6\u662f\u7528\u4e8e\u8bad\u7ec3\u57fa\u4e8e\u59ff\u6001\u6a21\u578b\u7684YOLO&#xff08;You Only Look Once&#xff09;\u68c0\u6d4b\u5668\u7684&#xff0c;\u5c5e\u4e8eUltralytics YOLO\u6846\u67b6\u7684\u4e00\u90e8\u5206\u3002\u5b83\u6269\u5c55\u4e86YOLO\u7684\u68c0\u6d4b\u8bad\u7ec3\u5668\u7c7b&#xff0c;\u4e13\u95e8\u7528\u4e8e\u59ff\u6001\u4f30\u8ba1\u4efb\u52a1\u3002<\/li>\n<p>\u9996\u5148&#xff0c;\u6587\u4ef6\u5bfc\u5165\u4e86\u4e00\u4e9b\u5fc5\u8981\u7684\u6a21\u5757\u548c\u7c7b&#xff0c;\u5305\u62ecYOLO\u6a21\u578b\u3001PoseModel&#xff08;\u59ff\u6001\u6a21\u578b&#xff09;\u3001\u9ed8\u8ba4\u914d\u7f6e\u3001\u65e5\u5fd7\u8bb0\u5f55\u5668\u4ee5\u53ca\u7528\u4e8e\u7ed8\u56fe\u7684\u5de5\u5177\u3002PoseTrainer\u7c7b\u7ee7\u627f\u81eayolo.detect.DetectionTrainer\u7c7b&#xff0c;\u63d0\u4f9b\u4e86\u8bad\u7ec3\u59ff\u6001\u6a21\u578b\u7684\u529f\u80fd\u3002<\/p>\n<p>\u5728PoseTrainer\u7c7b\u7684\u6784\u9020\u51fd\u6570\u4e2d&#xff0c;\u521d\u59cb\u5316\u4e86\u8bad\u7ec3\u5668\u7684\u914d\u7f6e\u548c\u53c2\u6570\u3002\u5982\u679c\u6ca1\u6709\u63d0\u4f9b\u8986\u76d6\u53c2\u6570&#xff0c;\u5219\u9ed8\u8ba4\u521b\u5efa\u4e00\u4e2a\u7a7a\u5b57\u5178&#xff0c;\u5e76\u5c06\u4efb\u52a1\u7c7b\u578b\u8bbe\u7f6e\u4e3a\u2019pose\u2019\u3002\u5982\u679c\u8bbe\u5907\u88ab\u8bbe\u7f6e\u4e3a\u2019MPS\u2019&#xff08;Apple\u7684\u91d1\u5c5e\u6027\u80fd\u7740\u8272\u5668&#xff09;&#xff0c;\u5219\u4f1a\u53d1\u51fa\u8b66\u544a&#xff0c;\u5efa\u8bae\u4f7f\u7528\u2019cpu\u2019\u8bbe\u5907\u8fdb\u884c\u59ff\u6001\u6a21\u578b\u7684\u8bad\u7ec3&#xff0c;\u4ee5\u907f\u514d\u5df2\u77e5\u7684Bug\u3002<\/p>\n<p>get_model\u65b9\u6cd5\u7528\u4e8e\u83b7\u53d6\u59ff\u6001\u4f30\u8ba1\u6a21\u578b\u3002\u5b83\u6839\u636e\u7ed9\u5b9a\u7684\u914d\u7f6e\u548c\u6743\u91cd\u521d\u59cb\u5316PoseModel\u5b9e\u4f8b&#xff0c;\u5e76\u5728\u63d0\u4f9b\u6743\u91cd\u7684\u60c5\u51b5\u4e0b\u52a0\u8f7d\u6a21\u578b\u6743\u91cd\u3002set_model_attributes\u65b9\u6cd5\u5219\u8bbe\u7f6ePoseModel\u7684\u5173\u952e\u70b9\u5f62\u72b6\u5c5e\u6027&#xff0c;\u4ee5\u786e\u4fdd\u6a21\u578b\u80fd\u591f\u6b63\u786e\u5904\u7406\u8f93\u5165\u6570\u636e\u3002<\/p>\n<p>get_validator\u65b9\u6cd5\u8fd4\u56de\u4e00\u4e2aPoseValidator\u5b9e\u4f8b&#xff0c;\u7528\u4e8e\u6a21\u578b\u9a8c\u8bc1\u3002\u5b83\u8bbe\u7f6e\u4e86\u635f\u5931\u540d\u79f0&#xff0c;\u5305\u62ec\u8fb9\u754c\u6846\u635f\u5931\u3001\u59ff\u6001\u635f\u5931\u3001\u5173\u952e\u5bf9\u8c61\u635f\u5931\u3001\u7c7b\u522b\u635f\u5931\u548c\u5206\u5e03\u5f0f\u7126\u70b9\u635f\u5931\u3002plot_training_samples\u65b9\u6cd5\u7528\u4e8e\u7ed8\u5236\u4e00\u6279\u8bad\u7ec3\u6837\u672c&#xff0c;\u663e\u793a\u5e26\u6709\u6ce8\u91ca\u7684\u7c7b\u6807\u7b7e\u3001\u8fb9\u754c\u6846\u548c\u5173\u952e\u70b9&#xff0c;\u4fbf\u4e8e\u53ef\u89c6\u5316\u8bad\u7ec3\u8fc7\u7a0b\u4e2d\u7684\u6570\u636e\u3002\u6700\u540e&#xff0c;plot_metrics\u65b9\u6cd5\u7528\u4e8e\u7ed8\u5236\u8bad\u7ec3\u548c\u9a8c\u8bc1\u7684\u6307\u6807&#xff0c;\u5e76\u4fdd\u5b58\u7ed3\u679c\u56fe\u50cf\u3002<\/p>\n<p>\u603b\u4f53\u800c\u8a00&#xff0c;\u8fd9\u4e2a\u6587\u4ef6\u63d0\u4f9b\u4e86\u4e00\u4e2a\u5b8c\u6574\u7684\u6846\u67b6&#xff0c;\u7528\u4e8e\u8bad\u7ec3\u59ff\u6001\u4f30\u8ba1\u6a21\u578b&#xff0c;\u652f\u6301\u6570\u636e\u53ef\u89c6\u5316\u548c\u6027\u80fd\u8bc4\u4f30\u3002<\/p>\n<h2>\u6e90\u7801\u6587\u4ef6<\/h2>\n<p><img decoding=\"async\" src=\"https:\/\/www.wsisp.com\/helps\/wp-content\/uploads\/2026\/02\/20260226005108-699f98fc1280c.png\" alt=\"\u5728\u8fd9\u91cc\u63d2\u5165\u56fe\u7247\u63cf\u8ff0\" \/><\/p>\n<h2>\u6e90\u7801\u83b7\u53d6<\/h2>\n<p>\u6b22\u8fce\u5927\u5bb6\u70b9\u8d5e\u3001\u6536\u85cf\u3001\u5173\u6ce8\u3001\u8bc4\u8bba\u5566 \u3001\u67e5\u770b&#x1f447;&#x1f3fb;\u83b7\u53d6\u8054\u7cfb\u65b9\u5f0f&#x1f447;&#x1f3fb;<\/p>\n","protected":false},"excerpt":{"rendered":"<p>\u80cc\u666f\u610f\u4e49<br \/>\n\u968f\u7740\u5168\u7403\u57ce\u9547\u5316\u8fdb\u7a0b\u7684\u52a0\u901f&#xff0c;\u4e61\u6751\u5730\u533a\u7684\u751f\u6001\u73af\u5883\u548c\u4ea4\u901a\u57fa\u7840\u8bbe\u65bd\u9762\u4e34\u7740\u524d\u6240\u672a\u6709\u7684\u6311\u6218\u3002\u4e61\u6751\u5c0f\u9053\u4f5c\u4e3a\u8fde\u63a5\u6751\u5e84\u4e0e\u5916\u754c\u7684\u91cd\u8981\u901a\u9053&#xff0c;\u4e0d\u4ec5\u627f\u8f7d\u7740\u519c\u6c11\u7684\u65e5\u5e38\u51fa\u884c\u548c\u519c\u4ea7\u54c1\u8fd0\u8f93&#xff0c;\u4e5f\u5728\u4e61\u6751\u65c5\u6e38\u548c\u751f\u6001\u4fdd\u62a4\u4e2d\u53d1\u6325\u7740\u91cd\u8981\u4f5c\u7528\u3002\u7136\u800c&#xff0c;\u4f20\u7edf\u7684\u4e61\u6751\u5c0f\u9053\u5f80\u5f80\u7f3a\u4e4f\u6709\u6548\u7684\u7ba1\u7406\u548c\u7ef4\u62a4&#xff0c;\u5bfc\u81f4\u9053\u8def\u635f\u574f\u3001\u690d\u88ab\u8fc7\u5ea6\u751f\u957f\u7b49\u95ee\u9898&#xff0c;\u4e25\u91cd\u5f71\u54cd\u4e86\u4e61\u6751\u7684\u53ef\u6301\u7eed\u53d1\u5c55\u3002\u56e0\u6b64&#xff0c;\u5982\u4f55\u6709\u6548\u5730\u76d1\u6d4b\u548c\u7ba1\u7406\u4e61\u6751\u5c0f\u9053&#xff0c;\u6210\u4e3a\u4e86\u5f53\u524d\u7814\u7a76\u7684\u70ed<\/p>\n","protected":false},"author":2,"featured_media":78119,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[1],"tags":[8603,156,87,5134,523],"topic":[],"class_list":["post-78128","post","type-post","status-publish","format-standard","has-post-thumbnail","hentry","category-server","tag-8603","tag-yolo","tag-87","tag-5134","tag-523"],"yoast_head":"<!-- This site is optimized with the Yoast SEO plugin v20.3 - https:\/\/yoast.com\/wordpress\/plugins\/seo\/ -->\n<title>\u3010\u5b8c\u6574\u6e90\u7801+\u6570\u636e\u96c6+\u90e8\u7f72\u6559\u7a0b\u3011\u4e61\u6751\u5c0f\u9053\u56fe\u50cf\u5206\u5272\u7cfb\u7edf\u6e90\u7801\uff06\u6570\u636e\u96c6\u5206\u4eab [yolov8-seg-C2f-Faster-EMA\uff06yolov8-seg-HGNetV2\u7b4950+\u5168\u5957\u6539\u8fdb\u521b\u65b0\u70b9\u53d1\u520a_\u4e00\u952e\u8bad\u7ec3\u6559\u7a0b - \u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3<\/title>\n<meta name=\"robots\" content=\"index, follow, max-snippet:-1, max-image-preview:large, max-video-preview:-1\" \/>\n<link rel=\"canonical\" href=\"https:\/\/www.wsisp.com\/helps\/78128.html\" \/>\n<meta property=\"og:locale\" content=\"zh_CN\" \/>\n<meta property=\"og:type\" content=\"article\" \/>\n<meta property=\"og:title\" content=\"\u3010\u5b8c\u6574\u6e90\u7801+\u6570\u636e\u96c6+\u90e8\u7f72\u6559\u7a0b\u3011\u4e61\u6751\u5c0f\u9053\u56fe\u50cf\u5206\u5272\u7cfb\u7edf\u6e90\u7801\uff06\u6570\u636e\u96c6\u5206\u4eab [yolov8-seg-C2f-Faster-EMA\uff06yolov8-seg-HGNetV2\u7b4950+\u5168\u5957\u6539\u8fdb\u521b\u65b0\u70b9\u53d1\u520a_\u4e00\u952e\u8bad\u7ec3\u6559\u7a0b - \u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3\" \/>\n<meta property=\"og:description\" content=\"\u80cc\u666f\u610f\u4e49 \u968f\u7740\u5168\u7403\u57ce\u9547\u5316\u8fdb\u7a0b\u7684\u52a0\u901f&#xff0c;\u4e61\u6751\u5730\u533a\u7684\u751f\u6001\u73af\u5883\u548c\u4ea4\u901a\u57fa\u7840\u8bbe\u65bd\u9762\u4e34\u7740\u524d\u6240\u672a\u6709\u7684\u6311\u6218\u3002\u4e61\u6751\u5c0f\u9053\u4f5c\u4e3a\u8fde\u63a5\u6751\u5e84\u4e0e\u5916\u754c\u7684\u91cd\u8981\u901a\u9053&#xff0c;\u4e0d\u4ec5\u627f\u8f7d\u7740\u519c\u6c11\u7684\u65e5\u5e38\u51fa\u884c\u548c\u519c\u4ea7\u54c1\u8fd0\u8f93&#xff0c;\u4e5f\u5728\u4e61\u6751\u65c5\u6e38\u548c\u751f\u6001\u4fdd\u62a4\u4e2d\u53d1\u6325\u7740\u91cd\u8981\u4f5c\u7528\u3002\u7136\u800c&#xff0c;\u4f20\u7edf\u7684\u4e61\u6751\u5c0f\u9053\u5f80\u5f80\u7f3a\u4e4f\u6709\u6548\u7684\u7ba1\u7406\u548c\u7ef4\u62a4&#xff0c;\u5bfc\u81f4\u9053\u8def\u635f\u574f\u3001\u690d\u88ab\u8fc7\u5ea6\u751f\u957f\u7b49\u95ee\u9898&#xff0c;\u4e25\u91cd\u5f71\u54cd\u4e86\u4e61\u6751\u7684\u53ef\u6301\u7eed\u53d1\u5c55\u3002\u56e0\u6b64&#xff0c;\u5982\u4f55\u6709\u6548\u5730\u76d1\u6d4b\u548c\u7ba1\u7406\u4e61\u6751\u5c0f\u9053&#xff0c;\u6210\u4e3a\u4e86\u5f53\u524d\u7814\u7a76\u7684\u70ed\" \/>\n<meta property=\"og:url\" content=\"https:\/\/www.wsisp.com\/helps\/78128.html\" \/>\n<meta property=\"og:site_name\" content=\"\u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3\" \/>\n<meta property=\"article:published_time\" content=\"2026-02-26T00:51:09+00:00\" \/>\n<meta property=\"og:image\" content=\"https:\/\/www.wsisp.com\/helps\/wp-content\/uploads\/2026\/02\/20260226005101-699f98f5b59df.jpg\" \/>\n<meta name=\"author\" content=\"admin\" \/>\n<meta name=\"twitter:card\" content=\"summary_large_image\" \/>\n<meta name=\"twitter:label1\" content=\"\u4f5c\u8005\" \/>\n\t<meta name=\"twitter:data1\" content=\"admin\" \/>\n\t<meta name=\"twitter:label2\" content=\"\u9884\u8ba1\u9605\u8bfb\u65f6\u95f4\" \/>\n\t<meta name=\"twitter:data2\" content=\"13 \u5206\" \/>\n<script type=\"application\/ld+json\" class=\"yoast-schema-graph\">{\"@context\":\"https:\/\/schema.org\",\"@graph\":[{\"@type\":\"WebPage\",\"@id\":\"https:\/\/www.wsisp.com\/helps\/78128.html\",\"url\":\"https:\/\/www.wsisp.com\/helps\/78128.html\",\"name\":\"\u3010\u5b8c\u6574\u6e90\u7801+\u6570\u636e\u96c6+\u90e8\u7f72\u6559\u7a0b\u3011\u4e61\u6751\u5c0f\u9053\u56fe\u50cf\u5206\u5272\u7cfb\u7edf\u6e90\u7801\uff06\u6570\u636e\u96c6\u5206\u4eab [yolov8-seg-C2f-Faster-EMA\uff06yolov8-seg-HGNetV2\u7b4950+\u5168\u5957\u6539\u8fdb\u521b\u65b0\u70b9\u53d1\u520a_\u4e00\u952e\u8bad\u7ec3\u6559\u7a0b - \u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3\",\"isPartOf\":{\"@id\":\"https:\/\/www.wsisp.com\/helps\/#website\"},\"datePublished\":\"2026-02-26T00:51:09+00:00\",\"dateModified\":\"2026-02-26T00:51:09+00:00\",\"author\":{\"@id\":\"https:\/\/www.wsisp.com\/helps\/#\/schema\/person\/358e386c577a3ab51c4493330a20ad41\"},\"breadcrumb\":{\"@id\":\"https:\/\/www.wsisp.com\/helps\/78128.html#breadcrumb\"},\"inLanguage\":\"zh-Hans\",\"potentialAction\":[{\"@type\":\"ReadAction\",\"target\":[\"https:\/\/www.wsisp.com\/helps\/78128.html\"]}]},{\"@type\":\"BreadcrumbList\",\"@id\":\"https:\/\/www.wsisp.com\/helps\/78128.html#breadcrumb\",\"itemListElement\":[{\"@type\":\"ListItem\",\"position\":1,\"name\":\"\u9996\u9875\",\"item\":\"https:\/\/www.wsisp.com\/helps\"},{\"@type\":\"ListItem\",\"position\":2,\"name\":\"\u3010\u5b8c\u6574\u6e90\u7801+\u6570\u636e\u96c6+\u90e8\u7f72\u6559\u7a0b\u3011\u4e61\u6751\u5c0f\u9053\u56fe\u50cf\u5206\u5272\u7cfb\u7edf\u6e90\u7801\uff06\u6570\u636e\u96c6\u5206\u4eab [yolov8-seg-C2f-Faster-EMA\uff06yolov8-seg-HGNetV2\u7b4950+\u5168\u5957\u6539\u8fdb\u521b\u65b0\u70b9\u53d1\u520a_\u4e00\u952e\u8bad\u7ec3\u6559\u7a0b\"}]},{\"@type\":\"WebSite\",\"@id\":\"https:\/\/www.wsisp.com\/helps\/#website\",\"url\":\"https:\/\/www.wsisp.com\/helps\/\",\"name\":\"\u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3\",\"description\":\"\u9999\u6e2f\u670d\u52a1\u5668_\u9999\u6e2f\u4e91\u670d\u52a1\u5668\u8d44\u8baf_\u670d\u52a1\u5668\u5e2e\u52a9\u6587\u6863_\u670d\u52a1\u5668\u6559\u7a0b\",\"potentialAction\":[{\"@type\":\"SearchAction\",\"target\":{\"@type\":\"EntryPoint\",\"urlTemplate\":\"https:\/\/www.wsisp.com\/helps\/?s={search_term_string}\"},\"query-input\":\"required name=search_term_string\"}],\"inLanguage\":\"zh-Hans\"},{\"@type\":\"Person\",\"@id\":\"https:\/\/www.wsisp.com\/helps\/#\/schema\/person\/358e386c577a3ab51c4493330a20ad41\",\"name\":\"admin\",\"image\":{\"@type\":\"ImageObject\",\"inLanguage\":\"zh-Hans\",\"@id\":\"https:\/\/www.wsisp.com\/helps\/#\/schema\/person\/image\/\",\"url\":\"https:\/\/gravatar.wp-china-yes.net\/avatar\/?s=96&d=mystery\",\"contentUrl\":\"https:\/\/gravatar.wp-china-yes.net\/avatar\/?s=96&d=mystery\",\"caption\":\"admin\"},\"sameAs\":[\"http:\/\/wp.wsisp.com\"],\"url\":\"https:\/\/www.wsisp.com\/helps\/author\/admin\"}]}<\/script>\n<!-- \/ Yoast SEO plugin. -->","yoast_head_json":{"title":"\u3010\u5b8c\u6574\u6e90\u7801+\u6570\u636e\u96c6+\u90e8\u7f72\u6559\u7a0b\u3011\u4e61\u6751\u5c0f\u9053\u56fe\u50cf\u5206\u5272\u7cfb\u7edf\u6e90\u7801\uff06\u6570\u636e\u96c6\u5206\u4eab [yolov8-seg-C2f-Faster-EMA\uff06yolov8-seg-HGNetV2\u7b4950+\u5168\u5957\u6539\u8fdb\u521b\u65b0\u70b9\u53d1\u520a_\u4e00\u952e\u8bad\u7ec3\u6559\u7a0b - \u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3","robots":{"index":"index","follow":"follow","max-snippet":"max-snippet:-1","max-image-preview":"max-image-preview:large","max-video-preview":"max-video-preview:-1"},"canonical":"https:\/\/www.wsisp.com\/helps\/78128.html","og_locale":"zh_CN","og_type":"article","og_title":"\u3010\u5b8c\u6574\u6e90\u7801+\u6570\u636e\u96c6+\u90e8\u7f72\u6559\u7a0b\u3011\u4e61\u6751\u5c0f\u9053\u56fe\u50cf\u5206\u5272\u7cfb\u7edf\u6e90\u7801\uff06\u6570\u636e\u96c6\u5206\u4eab [yolov8-seg-C2f-Faster-EMA\uff06yolov8-seg-HGNetV2\u7b4950+\u5168\u5957\u6539\u8fdb\u521b\u65b0\u70b9\u53d1\u520a_\u4e00\u952e\u8bad\u7ec3\u6559\u7a0b - \u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3","og_description":"\u80cc\u666f\u610f\u4e49 \u968f\u7740\u5168\u7403\u57ce\u9547\u5316\u8fdb\u7a0b\u7684\u52a0\u901f&#xff0c;\u4e61\u6751\u5730\u533a\u7684\u751f\u6001\u73af\u5883\u548c\u4ea4\u901a\u57fa\u7840\u8bbe\u65bd\u9762\u4e34\u7740\u524d\u6240\u672a\u6709\u7684\u6311\u6218\u3002\u4e61\u6751\u5c0f\u9053\u4f5c\u4e3a\u8fde\u63a5\u6751\u5e84\u4e0e\u5916\u754c\u7684\u91cd\u8981\u901a\u9053&#xff0c;\u4e0d\u4ec5\u627f\u8f7d\u7740\u519c\u6c11\u7684\u65e5\u5e38\u51fa\u884c\u548c\u519c\u4ea7\u54c1\u8fd0\u8f93&#xff0c;\u4e5f\u5728\u4e61\u6751\u65c5\u6e38\u548c\u751f\u6001\u4fdd\u62a4\u4e2d\u53d1\u6325\u7740\u91cd\u8981\u4f5c\u7528\u3002\u7136\u800c&#xff0c;\u4f20\u7edf\u7684\u4e61\u6751\u5c0f\u9053\u5f80\u5f80\u7f3a\u4e4f\u6709\u6548\u7684\u7ba1\u7406\u548c\u7ef4\u62a4&#xff0c;\u5bfc\u81f4\u9053\u8def\u635f\u574f\u3001\u690d\u88ab\u8fc7\u5ea6\u751f\u957f\u7b49\u95ee\u9898&#xff0c;\u4e25\u91cd\u5f71\u54cd\u4e86\u4e61\u6751\u7684\u53ef\u6301\u7eed\u53d1\u5c55\u3002\u56e0\u6b64&#xff0c;\u5982\u4f55\u6709\u6548\u5730\u76d1\u6d4b\u548c\u7ba1\u7406\u4e61\u6751\u5c0f\u9053&#xff0c;\u6210\u4e3a\u4e86\u5f53\u524d\u7814\u7a76\u7684\u70ed","og_url":"https:\/\/www.wsisp.com\/helps\/78128.html","og_site_name":"\u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3","article_published_time":"2026-02-26T00:51:09+00:00","og_image":[{"url":"https:\/\/www.wsisp.com\/helps\/wp-content\/uploads\/2026\/02\/20260226005101-699f98f5b59df.jpg"}],"author":"admin","twitter_card":"summary_large_image","twitter_misc":{"\u4f5c\u8005":"admin","\u9884\u8ba1\u9605\u8bfb\u65f6\u95f4":"13 \u5206"},"schema":{"@context":"https:\/\/schema.org","@graph":[{"@type":"WebPage","@id":"https:\/\/www.wsisp.com\/helps\/78128.html","url":"https:\/\/www.wsisp.com\/helps\/78128.html","name":"\u3010\u5b8c\u6574\u6e90\u7801+\u6570\u636e\u96c6+\u90e8\u7f72\u6559\u7a0b\u3011\u4e61\u6751\u5c0f\u9053\u56fe\u50cf\u5206\u5272\u7cfb\u7edf\u6e90\u7801\uff06\u6570\u636e\u96c6\u5206\u4eab [yolov8-seg-C2f-Faster-EMA\uff06yolov8-seg-HGNetV2\u7b4950+\u5168\u5957\u6539\u8fdb\u521b\u65b0\u70b9\u53d1\u520a_\u4e00\u952e\u8bad\u7ec3\u6559\u7a0b - \u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3","isPartOf":{"@id":"https:\/\/www.wsisp.com\/helps\/#website"},"datePublished":"2026-02-26T00:51:09+00:00","dateModified":"2026-02-26T00:51:09+00:00","author":{"@id":"https:\/\/www.wsisp.com\/helps\/#\/schema\/person\/358e386c577a3ab51c4493330a20ad41"},"breadcrumb":{"@id":"https:\/\/www.wsisp.com\/helps\/78128.html#breadcrumb"},"inLanguage":"zh-Hans","potentialAction":[{"@type":"ReadAction","target":["https:\/\/www.wsisp.com\/helps\/78128.html"]}]},{"@type":"BreadcrumbList","@id":"https:\/\/www.wsisp.com\/helps\/78128.html#breadcrumb","itemListElement":[{"@type":"ListItem","position":1,"name":"\u9996\u9875","item":"https:\/\/www.wsisp.com\/helps"},{"@type":"ListItem","position":2,"name":"\u3010\u5b8c\u6574\u6e90\u7801+\u6570\u636e\u96c6+\u90e8\u7f72\u6559\u7a0b\u3011\u4e61\u6751\u5c0f\u9053\u56fe\u50cf\u5206\u5272\u7cfb\u7edf\u6e90\u7801\uff06\u6570\u636e\u96c6\u5206\u4eab [yolov8-seg-C2f-Faster-EMA\uff06yolov8-seg-HGNetV2\u7b4950+\u5168\u5957\u6539\u8fdb\u521b\u65b0\u70b9\u53d1\u520a_\u4e00\u952e\u8bad\u7ec3\u6559\u7a0b"}]},{"@type":"WebSite","@id":"https:\/\/www.wsisp.com\/helps\/#website","url":"https:\/\/www.wsisp.com\/helps\/","name":"\u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3","description":"\u9999\u6e2f\u670d\u52a1\u5668_\u9999\u6e2f\u4e91\u670d\u52a1\u5668\u8d44\u8baf_\u670d\u52a1\u5668\u5e2e\u52a9\u6587\u6863_\u670d\u52a1\u5668\u6559\u7a0b","potentialAction":[{"@type":"SearchAction","target":{"@type":"EntryPoint","urlTemplate":"https:\/\/www.wsisp.com\/helps\/?s={search_term_string}"},"query-input":"required name=search_term_string"}],"inLanguage":"zh-Hans"},{"@type":"Person","@id":"https:\/\/www.wsisp.com\/helps\/#\/schema\/person\/358e386c577a3ab51c4493330a20ad41","name":"admin","image":{"@type":"ImageObject","inLanguage":"zh-Hans","@id":"https:\/\/www.wsisp.com\/helps\/#\/schema\/person\/image\/","url":"https:\/\/gravatar.wp-china-yes.net\/avatar\/?s=96&d=mystery","contentUrl":"https:\/\/gravatar.wp-china-yes.net\/avatar\/?s=96&d=mystery","caption":"admin"},"sameAs":["http:\/\/wp.wsisp.com"],"url":"https:\/\/www.wsisp.com\/helps\/author\/admin"}]}},"_links":{"self":[{"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/posts\/78128","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/users\/2"}],"replies":[{"embeddable":true,"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/comments?post=78128"}],"version-history":[{"count":0,"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/posts\/78128\/revisions"}],"wp:featuredmedia":[{"embeddable":true,"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/media\/78119"}],"wp:attachment":[{"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/media?parent=78128"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/categories?post=78128"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/tags?post=78128"},{"taxonomy":"topic","embeddable":true,"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/topic?post=78128"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}