我有一个小的脚本,用于检查我的对象是否在图像中。如果存在,则将图像写入文件夹。内部有多个子文件夹。写入时,第一个文件夹的数据可以正常工作,但是当数据写入第二个子文件夹时,第一个文件夹的数据也将附加-带有第二个文件夹的数据
除了将数据写入磁盘时,第一个子文件夹的数据也被附加到第二个子文件夹的数据的问题之外,代码工作得很好。下面是我的代码
def target_non_target(input_frames_folder,model_file,output):
if not os.path.exists(output):
os.makedirs(output,exist_ok=True)
count=0
folders = glob(input_frames_folder)
img_list = []
for folder in folders:
folder_name=os.path.basename(folder)
#print(folder_name)
out_path = output +"\\" + folder_name
print(out_path)
os.makedirs(out_path,exist_ok=True)
for f in glob(folder+"/*.jpg"):
img_list.append(f)
for i in range(len(img_list)):
v1=os.path.basename(img_list[i])
img_name = os.path.splitext(v1)[0]
image = cv2.imread(img_list[i])
orig = image.copy()
image = cv2.resize(image, (28, 28))
image = image.astype("float") / 255.0
image = img_to_array(image)
image = np.expand_dims(image, axis=0)
print("[INFO] loading network...")
model = load_model(model_file)
(non_target, target) = model.predict(image)[0]
if target > non_target:
label = "Target"
else:
label = "Non Target"
probab = target if target > non_target else non_target
label = "{}: {:.2f}%".format(label, probab * 100)
op = imutils.resize(orig, width=400)
cv2.putText(op, label, (10, 25), cv2.FONT_HERSHEY_SIMPLEX,0.7, (0, 255, 0), 2)
if target > non_target:
cv2.imwrite(out_path+"/"+"\\{}.jpg".format(img_name),orig)
cv2.waitKey(0)
#return target_op
frames_folder = ("C:\\Python36\\videos\\videos_new\\*")
model = ("C:\\Python35\\target_non_target\\target_non_target.model")
output_folder = ("C:\\Python35\\target_non_target\\Target_images_new\\")
target_check = target_non_target(frames_folder,model,output_folder)
假设在主文件夹X内有两个子文件夹A和B,会有更多的子文件夹。在将输出写入磁盘时,A的数据被完美地写入了,而在B的数据写入了文件夹的数据时A和文件夹B都将附加到B文件夹中。我希望数据位于它们各自的文件夹中。知道可以在我的脚本中进行哪些更改以获取所需的输出
答案 0 :(得分:1)
您正在使用await tester.pumpWidget(Login());
进行初始化,但是您需要在每个文件夹循环的末尾重复此操作以将其重置为空。现在,您将保留所有结果,然后,当您移至下一个文件夹时,将保留先前的结果并添加。
img_list = []
循环的最后需要有for folder in folders
更新后的完整代码:
img_list = []