{"id":1959,"date":"2024-05-03T19:19:25","date_gmt":"2024-05-03T19:19:25","guid":{"rendered":"https:\/\/www.nicekj.com\/?p=1959"},"modified":"2024-05-03T19:19:25","modified_gmt":"2024-05-03T19:19:25","slug":"yolov8moxingxunlianchuxuezhebijifanwaipianyijianzhuanyilabalimgbiaoqianwenjian","status":"publish","type":"post","link":"https:\/\/www.nicekj.com\/yolov8moxingxunlianchuxuezhebijifanwaipianyijianzhuanyilabalimgbiaoqianwenjian.html","title":{"rendered":"Yolo V8 \u6a21\u578b\u8bad\u7ec3\u521d\u5b66\u8005\u7b14\u8bb0\uff08\u756a\u5916\u7bc7\uff09&#8211; \u4e00\u952e\u8f6c\u79fblabalimg\u6807\u7b7e\u6587\u4ef6"},"content":{"rendered":"<p>\u8854\u63a5\u4e0a\u4e00\u7bc7\uff1a<a href=\"https:\/\/juejin.cn\/post\/7344573755643101238\" title=\"https:\/\/juejin.cn\/post\/7344573755643101238\" target=\"_blank\" rel=\"noopener\"># YOLO V8 \u6a21\u578b\u8bad\u7ec3\u548c\u76ee\u6807\u68c0\u6d4b\u521d\u5b66\u8005\u7b14\u8bb0<\/a><\/p>\n<h2 data-id=\"heading-0\">\u524d\u8a00<\/h2>\n<ul>\n<li>\n<p>\u4e0a\u4e00\u7bc7\u6587\u7ae0\u4e2d\uff0c\u6211\u7528 labelimg \u5de5\u5177\u4e3a\u6bcf\u5f20\u8bad\u7ec3\u6240\u9700\u56fe\u7247\u6253\u6807\u7b7e(\u6807\u6ce8)\u540e\uff0c\u4f1a\u4fdd\u5b58\u8f93\u51fa\u5bf9\u5e94\u5230\u6bcf\u4e00\u5f20\u56fe\u7247\u7684 xxx.txt \u6587\u672c\u548cclassess.txt\uff08\u8bb0\u5f55\u6807\u7b7e\u540d\u7684\u6587\u672c\uff09\uff0c\u8fd9\u4e9b xxx.txt \u6587\u672c\u8bb0\u5f55\u7740\u5bf9\u5e94\u6bcf\u4e00\u5f20\u56fe\u7247\u4e0a\u7684\u6807\u6ce8\u7d22\u5f15\u548c\u6807\u6ce8\u5750\u6807\u4fe1\u606f<\/p>\n<\/li>\n<li>\n<p>\u5e76\u4e14\u5728\u6267\u884c\u8bad\u7ec3\u811a\u672c\u4e4b\u524d\uff0c\u6211\u4eec\u9700\u8981\u628a\u6253\u6807\u7b7e\u7684\u539f\u59cb\u56fe\u7247\u548c\u6807\u7b7e\u6587\u672c\u6574\u7406\u5230 data\/xxx \u76ee\u5f55\u4e0b\uff0c\u8fd8\u9700\u8981\u5728 data\/xxx.yaml \u4e2d\u5199\u597d\u914d\u7f6e\u4fe1\u606f<\/p>\n<\/li>\n<li>\n<p>\u90a3\u6bcf\u6b21\u8bad\u7ec3\u6a21\u578b\u4e4b\u524d\u90fd\u8981\u624b\u52a8\u8fd9\u4e48\u505a\u5c31\u592a\u8d39\u65f6\u95f4\u4e86\uff0c\u4e8e\u662f\u6211\u5199\u4e0b\u6765\u8fd9\u4e2a\u4e00\u52b3\u6c38\u9038\u7684 copy_labelimgfiles_to_datadir.py \u811a\u672c<\/p>\n<\/li>\n<li>\n<p>\u4ee5\u4e0b\u662f\u8bf4\u660e\u6848\u4f8b\u622a\u56fe<br>\n\n<figure class=\"wp-block-image size-large\"><img decoding=\"async\" src=\"https:\/\/p3-juejin.byteimg.com\/tos-cn-i-k3u1fbpfcp\/ea5bf2f5781748e4a418ef6b8dba72bd~tplv-k3u1fbpfcp-jj-mark:3024:0:0:0:q75.awebp#?w=255&amp;h=789&amp;s=19244&amp;e=png&amp;b=181818\" alt=\"\u5c4f\u5e55\u622a\u56fe 2024-04-14 111513.png\" \/><\/figure>\n<\/p>\n<\/li>\n<li>\n<p>\u5176\u4e2d\u81ea\u52a8\u5199\u5165\u7684 xxx.yaml \u6848\u4f8b\u622a\u56fe\u5982\u4e0b<br>\n\n<figure class=\"wp-block-image size-large\"><img decoding=\"async\" src=\"https:\/\/p6-juejin.byteimg.com\/tos-cn-i-k3u1fbpfcp\/a496b80067ed44b68b73f09a7036be9c~tplv-k3u1fbpfcp-jj-mark:3024:0:0:0:q75.awebp#?w=727&amp;h=148&amp;s=10373&amp;e=png&amp;b=181818\" alt=\"\u5c4f\u5e55\u622a\u56fe 2024-04-14 1124301.png\" \/><\/figure>\n<\/p>\n<\/li>\n<\/ul>\n<h2 data-id=\"heading-1\">\u811a\u672c\u6e90\u7801\u5728\u6b64\uff01\uff01\uff01<\/h2>\n<ul>\n<li>copy_labelimgfiles_to_datadir.py<\/li>\n<\/ul>\n<pre><\/div><div class=\"code-block-extension-headerRight\"><span class=\"code-block-extension-lang\">python<\/span><div class=\"code-block-extension-copyCodeBtn\">\u590d\u5236\u4ee3\u7801<\/div><\/div><\/div><code class=\"hljs language-python code-block-extension-codeShowNum\" lang=\"python\"><span class=\"code-block-extension-codeLine\" data-line-num=\"1\"><span class=\"hljs-keyword\">import<\/span> os<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"2\"><span class=\"hljs-keyword\">import<\/span> shutil<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"3\"><span class=\"hljs-keyword\">import<\/span> random<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"4\"><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"5\">cur_dir = os.path.dirname(__file__).replace(<span class=\"hljs-string\">'\\\\'<\/span>, <span class=\"hljs-string\">'\/'<\/span>)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"6\"><span class=\"hljs-built_in\">print<\/span>(cur_dir)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"7\"><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"8\">source_img_dir = cur_dir + <span class=\"hljs-string\">'\/source_img'<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"9\">labelimg_output_dir = cur_dir + <span class=\"hljs-string\">'\/labelimg_output'<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"10\"><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"11\">data_dir = cur_dir + <span class=\"hljs-string\">'\/data'<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"12\">data_images_dir = data_dir + <span class=\"hljs-string\">'\/images'<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"13\">data_labels_dir = data_dir + <span class=\"hljs-string\">'\/labels'<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"14\">data_train_dir = data_dir + <span class=\"hljs-string\">'\/train'<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"15\">data_train_images_dir = data_train_dir + <span class=\"hljs-string\">'\/images'<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"16\">data_train_labels_dir = data_train_dir + <span class=\"hljs-string\">'\/labels'<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"17\">data_val_dir = data_dir + <span class=\"hljs-string\">'\/val'<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"18\">data_val_images_dir = data_val_dir + <span class=\"hljs-string\">'\/images'<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"19\">data_val_labels_dir = data_val_dir + <span class=\"hljs-string\">'\/labels'<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"20\">data_test_dir = data_dir + <span class=\"hljs-string\">'\/test'<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"21\">data_test_images_dir = data_test_dir + <span class=\"hljs-string\">'\/images'<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"22\">data_test_labels_dir = data_test_dir + <span class=\"hljs-string\">'\/labels'<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"23\">data_yoloyaml_file_path = data_dir + <span class=\"hljs-string\">'\/yolov8nconfig.yaml'<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"24\"><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"25\"><span class=\"hljs-keyword\">def<\/span> <span class=\"hljs-title function_\">read_full_path_list<\/span>(<span class=\"hljs-params\">dir_path<\/span>):<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"26\">    full_path_list = []<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"27\">    files = os.listdir(dir_path)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"28\">    <span class=\"hljs-keyword\">for<\/span> f <span class=\"hljs-keyword\">in<\/span> files:<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"29\">        full_path_list.append(dir_path + <span class=\"hljs-string\">'\/'<\/span> + f)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"30\">    <span class=\"hljs-keyword\">return<\/span> full_path_list<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"31\"><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"32\"><span class=\"hljs-keyword\">def<\/span> <span class=\"hljs-title function_\">create_dir<\/span>(<span class=\"hljs-params\">dir_path<\/span>):<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"33\">    <span class=\"hljs-keyword\">if<\/span> <span class=\"hljs-keyword\">not<\/span> os.path.exists(dir_path):<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"34\">        os.mkdir(dir_path)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"35\"><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"36\"><span class=\"hljs-keyword\">def<\/span> <span class=\"hljs-title function_\">read_labelimg_files<\/span>():<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"37\">    <span class=\"hljs-comment\"># \u5224\u65ad\u8bfb\u53d6\u7684\u76ee\u5f55\u662f\u5426\u5b58\u5728<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"38\">    <span class=\"hljs-keyword\">if<\/span> <span class=\"hljs-keyword\">not<\/span> os.path.exists(source_img_dir):<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"39\">        <span class=\"hljs-built_in\">print<\/span>(<span class=\"hljs-string\">f'\u3010labelimg_output_dir\u3011: <span class=\"hljs-subst\">{source_img_dir}<\/span> not exists'<\/span>)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"40\">        <span class=\"hljs-keyword\">return<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"41\">    <\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"42\">    <span class=\"hljs-keyword\">if<\/span> <span class=\"hljs-keyword\">not<\/span> os.path.exists(labelimg_output_dir):<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"43\">        <span class=\"hljs-built_in\">print<\/span>(<span class=\"hljs-string\">f'\u3010labelimg_output_dir\u3011: <span class=\"hljs-subst\">{labelimg_output_dir}<\/span> not exists'<\/span>)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"44\">        <span class=\"hljs-keyword\">return<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"45\"><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"46\">    <span class=\"hljs-comment\"># \u8bfb\u53d6 source_img_dir \u76ee\u5f55\u4e0b\u7684\u6240\u6709\u6587\u4ef6<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"47\">    source_img_files_fullpath_list = read_full_path_list(source_img_dir)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"48\">    <span class=\"hljs-comment\"># \u8bfb\u53d6 labelimg_output_dir \u76ee\u5f55\u4e0b\u7684\u6240\u6709\u6587\u4ef6<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"49\">    labelimg_output_files_fullpath_list = read_full_path_list(labelimg_output_dir)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"50\">    labelimg_txt_files_fullpath_list = []<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"51\">    labelimg_classestxt_file_fullpath = <span class=\"hljs-string\">''<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"52\">    <span class=\"hljs-keyword\">for<\/span> f <span class=\"hljs-keyword\">in<\/span> labelimg_output_files_fullpath_list:<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"53\">        <span class=\"hljs-comment\"># \u63d0\u53d6\u51fa classes.txt \u7684\u6587\u4ef6<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"54\">        <span class=\"hljs-keyword\">if<\/span> f.endswith(<span class=\"hljs-string\">'classes.txt'<\/span>):<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"55\">            labelimg_classestxt_file_fullpath = f<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"56\">        <span class=\"hljs-comment\"># \u63d0\u53d6\u51fa txt \u6587\u4ef6<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"57\">        <span class=\"hljs-keyword\">elif<\/span> f.endswith(<span class=\"hljs-string\">'.txt'<\/span>):<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"58\">            labelimg_txt_files_fullpath_list.append(f)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"59\"><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"60\">    <span class=\"hljs-comment\"># print(f'source_img_files_fullpath_list: {source_img_files_fullpath_list}')<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"61\">    <span class=\"hljs-comment\"># print(f'labelimg_txt_files_fullpath_list: {labelimg_txt_files_fullpath_list}')<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"62\">    <span class=\"hljs-comment\"># print(f'labelimg_classestxt_file_fullpath: {labelimg_classestxt_file_fullpath}')<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"63\">    <\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"64\">    <span class=\"hljs-keyword\">return<\/span> source_img_files_fullpath_list, labelimg_txt_files_fullpath_list, labelimg_classestxt_file_fullpath<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"65\">    <\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"66\"><span class=\"hljs-keyword\">def<\/span> <span class=\"hljs-title function_\">copy_labelimg_files_to_data_dir<\/span>(<span class=\"hljs-params\">source_img_files_fullpath_list, labelimg_txt_files_fullpath_list, labelimg_classestxt_file_fullpath<\/span>):<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"67\">    <span class=\"hljs-comment\"># \u4ece labelimg_classestxt_file_fullpath \u6587\u4ef6\u4e2d\u9010\u884c\u8bfb\u53d6\u51fa\u7c7b\u540d<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"68\">    label_classes = []<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"69\">    <span class=\"hljs-keyword\">with<\/span> <span class=\"hljs-built_in\">open<\/span>(labelimg_classestxt_file_fullpath, <span class=\"hljs-string\">'r'<\/span>, encoding=<span class=\"hljs-string\">'utf-8'<\/span>) <span class=\"hljs-keyword\">as<\/span> f:<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"70\">        <span class=\"hljs-keyword\">for<\/span> line <span class=\"hljs-keyword\">in<\/span> f.readlines():<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"71\">            line = line.strip()<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"72\">            <span class=\"hljs-keyword\">if<\/span> line != <span class=\"hljs-string\">''<\/span>:<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"73\">                label_classes.append(line)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"74\">    <span class=\"hljs-comment\"># \u5199\u5165 yolo yaml \u914d\u7f6e\u6587\u4ef6<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"75\">    yoloymal_config_contents = [<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"76\">        <span class=\"hljs-string\">'train: '<\/span> + data_train_dir,<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"77\">        <span class=\"hljs-string\">'\\n'<\/span>,<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"78\">        <span class=\"hljs-string\">'val: '<\/span> + data_val_dir,<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"79\">        <span class=\"hljs-string\">'\\n'<\/span>,<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"80\">        <span class=\"hljs-string\">'test: '<\/span> + data_test_dir,<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"81\">        <span class=\"hljs-string\">'\\n'<\/span>,<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"82\">        <span class=\"hljs-string\">'\\n'<\/span>,<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"83\">        <span class=\"hljs-string\">'nc: '<\/span> + <span class=\"hljs-built_in\">str<\/span>(<span class=\"hljs-built_in\">len<\/span>(label_classes)),<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"84\">        <span class=\"hljs-string\">'\\n'<\/span>,<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"85\">        <span class=\"hljs-string\">'\\n'<\/span>,<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"86\">        <span class=\"hljs-string\">'names: '<\/span> + <span class=\"hljs-built_in\">str<\/span>(label_classes)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"87\">    ]<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"88\">    <span class=\"hljs-comment\"># \u6839\u636e yoloymal_config_contents \u6570\u7ec4\u4e2d\u7684\u5185\u5bb9\uff0c\u9010\u884c\u5199\u5165\u5230 data_yoloyaml_file_path \u6587\u4ef6\u4e2d<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"89\">    <span class=\"hljs-keyword\">with<\/span> <span class=\"hljs-built_in\">open<\/span>(data_yoloyaml_file_path, <span class=\"hljs-string\">'w'<\/span>, encoding=<span class=\"hljs-string\">'utf-8'<\/span>) <span class=\"hljs-keyword\">as<\/span> f:<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"90\">        f.writelines(yoloymal_config_contents)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"91\"><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"92\"><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"93\">    <span class=\"hljs-comment\"># \u628a\u53c2\u6570\u4e2d\u7684\u6587\u4ef6\u590d\u5236\u5230 data_dir \u76ee\u5f55\u4e0b\u5bf9\u5e94\u7684\u76ee\u5f55\u4e2d<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"94\">    <span class=\"hljs-keyword\">for<\/span> simg <span class=\"hljs-keyword\">in<\/span> source_img_files_fullpath_list:<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"95\">        shutil.copy2(simg, data_images_dir)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"96\">        shutil.copy2(simg, data_train_images_dir)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"97\"><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"98\">    <span class=\"hljs-keyword\">for<\/span> ltxt <span class=\"hljs-keyword\">in<\/span> labelimg_txt_files_fullpath_list:<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"99\">        shutil.copy2(ltxt, data_labels_dir)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"100\">        shutil.copy2(ltxt, data_train_labels_dir)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"101\"><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"102\">    <span class=\"hljs-comment\"># data\/val \u548c data\/test \u76ee\u5f55\u4e0b\u4e0d\u9700\u8981\u653e\u6240\u6709\u7684\u6587\u4ef6\uff0c\u53ea\u9700\u968f\u673a\u53d6\u4e00\u90e8\u5206\u7684\u6587\u4ef6\u5373\u53ef<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"103\">    simg_files_pre_parts = []<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"104\">    ltxt_files_pre_parts = []<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"105\">    simg_files_end_parts = []<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"106\">    ltxt_files_end_parts = []<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"107\">    <span class=\"hljs-comment\"># \u53d6\u6587\u4ef6\u5217\u8868\u4e00\u90e8\u5206\u7684\u51e0\u4e2a\u6587\u4ef6\uff0c\u653e\u5230 data\/val \u548c data\/test \u76ee\u5f55\u4e0b<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"108\">    files_pre_parts_i_list = []<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"109\">    files_end_parts_i_list = []<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"110\">    files_len = <span class=\"hljs-built_in\">len<\/span>(source_img_files_fullpath_list)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"111\">    half_files_len = <span class=\"hljs-built_in\">int<\/span>(files_len\/<span class=\"hljs-number\">2<\/span>)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"112\">    <span class=\"hljs-keyword\">for<\/span> i <span class=\"hljs-keyword\">in<\/span> <span class=\"hljs-built_in\">range<\/span>(half_files_len):<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"113\">        files_pre_parts_i_list.append(random.randint(<span class=\"hljs-number\">0<\/span>, half_files_len))<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"114\">        files_end_parts_i_list.append(random.randint(half_files_len+<span class=\"hljs-number\">1<\/span>, files_len-<span class=\"hljs-number\">1<\/span>))<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"115\">    <span class=\"hljs-keyword\">for<\/span> i <span class=\"hljs-keyword\">in<\/span> files_pre_parts_i_list:<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"116\">        simg_files_pre_parts.append(source_img_files_fullpath_list[i])<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"117\">        ltxt_files_pre_parts.append(labelimg_txt_files_fullpath_list[i])<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"118\">    <span class=\"hljs-keyword\">for<\/span> i <span class=\"hljs-keyword\">in<\/span> files_end_parts_i_list:<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"119\">        simg_files_end_parts.append(source_img_files_fullpath_list[i])<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"120\">        ltxt_files_end_parts.append(labelimg_txt_files_fullpath_list[i])<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"121\">    <span class=\"hljs-comment\"># \u628a\u524d\u9762\u968f\u673a\u63d0\u53d6\u51fa\u6765\u7684\u6587\u4ef6\uff0c\u5206\u522b\u590d\u5236\u5230 data\/val \u548c data\/test \u76ee\u5f55\u4e0b<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"122\">    <span class=\"hljs-keyword\">for<\/span> f <span class=\"hljs-keyword\">in<\/span> simg_files_pre_parts:<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"123\">            shutil.copy2(f, data_val_images_dir)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"124\">    <span class=\"hljs-keyword\">for<\/span> f <span class=\"hljs-keyword\">in<\/span> ltxt_files_pre_parts:<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"125\">            shutil.copy2(f, data_val_labels_dir)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"126\">    <span class=\"hljs-keyword\">for<\/span> f <span class=\"hljs-keyword\">in<\/span> simg_files_end_parts:<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"127\">            shutil.copy2(f, data_test_images_dir)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"128\">    <span class=\"hljs-keyword\">for<\/span> f <span class=\"hljs-keyword\">in<\/span> ltxt_files_end_parts:<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"129\">            shutil.copy2(f, data_test_labels_dir)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"130\">           <\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"131\"><span class=\"hljs-keyword\">def<\/span> <span class=\"hljs-title function_\">remove_data_dir<\/span>():<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"132\">    <span class=\"hljs-keyword\">if<\/span> os.path.exists(data_dir):<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"133\">        <span class=\"hljs-comment\"># \u5220\u9664 data_dir \u76ee\u5f55\u53ca\u5176\u4e0b\u9762\u7684\u6240\u6709\u5b50\u76ee\u5f55\u548c\u6587\u4ef6<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"134\">        shutil.rmtree(data_dir)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"135\"><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"136\"><span class=\"hljs-keyword\">def<\/span> <span class=\"hljs-title function_\">create_data_dir<\/span>():<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"137\">    remove_data_dir()<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"138\"><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"139\">    create_dir(data_dir)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"140\">    create_dir(data_images_dir)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"141\">    create_dir(data_labels_dir)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"142\">    create_dir(data_train_dir)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"143\">    create_dir(data_train_images_dir)    <\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"144\">    create_dir(data_train_labels_dir)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"145\">    create_dir(data_val_dir)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"146\">    create_dir(data_val_images_dir)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"147\">    create_dir(data_val_labels_dir)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"148\">    create_dir(data_test_dir)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"149\">    create_dir(data_test_images_dir)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"150\">    create_dir(data_test_labels_dir)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"151\"><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"152\"><span class=\"hljs-keyword\">if<\/span> __name__ == <span class=\"hljs-string\">'__main__'<\/span>:<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"153\">    source_img_files_fullpath_list, labelimg_txt_files_fullpath_list, labelimg_classestxt_file_fullpath = read_labelimg_files()<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"154\">    create_data_dir()<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"155\">    copy_labelimg_files_to_data_dir(source_img_files_fullpath_list, labelimg_txt_files_fullpath_list, labelimg_classestxt_file_fullpath)<\/span>\n<\/code><\/pre>","protected":false},"excerpt":{"rendered":"<p>\u5728\u6267\u884c\u8bad\u7ec3\u811a\u672c\u4e4b\u524d\uff0c\u6211\u4eec\u9700\u8981\u628a\u6253\u6807\u7b7e\u7684\u539f\u59cb\u56fe\u7247\u548c\u6807\u7b7e\u6587\u672c\u6574\u7406\u5230 data\/xxx \u76ee\u5f55\u4e0b\uff0c\u8fd8\u9700\u8981\u5728 data\/xxx.yaml \u4e2d\u5199\u597d\u914d\u7f6e\u4fe1\u606f \u90a3\u6bcf\u6b21\u8bad\u7ec3\u6a21\u578b\u4e4b\u524d\u90fd\u8981\u624b\u52a8\u8fd9\u4e48\u505a\u5c31\u592a\u8d39\u65f6\u95f4\u4e86\uff0c\u4e8e\u662f\u6211..<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"_acf_changed":false,"rank_math_title":"","rank_math_description":"","rank_math_focus_keyword":"","views":"","footnotes":""},"categories":[3],"tags":[136,126,127,128,129],"collection":[],"class_list":["post-1959","post","type-post","status-publish","format-standard","hentry","category-fenlei2","tag-136","tag-gpt","tag-ai","tag-128","tag-129"],"acf":[],"_links":{"self":[{"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/posts\/1959","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/comments?post=1959"}],"version-history":[{"count":0,"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/posts\/1959\/revisions"}],"wp:attachment":[{"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/media?parent=1959"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/categories?post=1959"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/tags?post=1959"},{"taxonomy":"collection","embeddable":true,"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/collection?post=1959"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}