电磁进入博客
这篇博客写的还不错,补充一点
怎么讲图片和标签划分成训练集和验证集
import numpy as np
import cv2
import shutil
with open("VOCdevkit/VOC2007/ImageSets/Main/train.txt") as f:
lines = f.readlines()
val_split = 0.3
np.random.seed(10101)
np.random.shuffle(lines)
print(type(lines))
np.random.seed(None)
num_val = int(len(lines) * val_split)
num_train = len(lines) - num_val
print(num_train)
print(len(lines))
# print(lines[])
# print('VOCdevkit/VOC2007/train/'+lines[1].strip('\n')+'.jpg')
# shutil.copy('VOCdevkit/VOC2007/labelYOLOs/1_1.txt','VOCdevkit/VOC2007/trainlabel/1_1.txt')
dirstr='VOCdevkit/VOC2007/JPEGImages/'
#
for itt in range(0,len(lines)-1):
#划分的测试集
if itt<num_train:
#读取图片并保存到其他目录
img = cv2.imread(dirstr+lines[itt].strip('\n')+'.jpg',
cv