1.3

spilt_train_val.py

@@ -30,14 +30,22 @@ file_test = open(txtsavepath + '/test.txt', 'w')
 file_train = open(txtsavepath + '/train.txt', 'w')
 file_val = open(txtsavepath + '/val.txt', 'w')
 
+addtrain_path = r"D:\PycharmProjects\Zhanting\yolov5_1\img_data\getimg_6.30"
 for i in list_index:
     name = total_xml[i][:-4] + '\n'
+    addimg_name = name.strip() + ".jpg"
+    # print(addimg_name,type(addimg_name),len(addimg_name))
     if i in trainval:
         file_trainval.write(name)
         if i in train:
             file_train.write(name)
         else:
-            file_val.write(name)
+            if addimg_name in os.listdir(addtrain_path):#把某些被划分到验证集的数据加入训练集中
+                print("addimg_name:",addimg_name)
+                file_train.write(name)
+            else:
+                file_val.write(name)
+            # file_val.write(name)
     else:
         file_test.write(name)