我正在尝试从一个子文件夹“input”(第10行)中提取图像文件并对其进行处理,然后将其保存到另一个子文件夹“output”(第10行),但处理后的文件未保存。但是如果我从保存代码的同一文件夹中获取所有图像文件(不在第10行的提取命令中写入输入),那么当我将其保存在子文件夹“output”中时,我就成功了。只要记住笔记;在这两种情况下,处理的文件都在显示并且是相同的,只是在一种情况下不保存。
import glob
import cv2
import numpy as np
import matplotlib.pyplot as plt
# initialize data and target as lists
data = []
target = []
# find all bmp files in the current folder
Files = glob.glob("input/*.bmp")
# go through all files and fit contour
for f in Files:
img = 255-cv2.imread(f,0) #0 for grayscale
white=cv2.imread("white.bmp")
# add image to the list
data.append(img)
# make lines thicker
kernel = np.ones((5,5),np.uint8)
img = cv2.dilate(img,kernel,iterations = 1)
# contour application
ret,thresh = cv2.threshold(img,127,255,0)
im2,contours,hierarchy = cv2.findContours(thresh, 1, 2)
# sort contours by area
areas = [cv2.contourArea(cnt) for cnt in contours]
idx = np.argsort(areas)[::-1]
contours = np.asarray(contours)[idx]
for cnt in contours:
(x,y),radius = cv2.minEnclosingCircle(cnt)
if radius < img.shape[0]-10 and radius > 20:
cv2.circle(white,(int(x),int(y)),int(radius),0,4)
break
plt.imshow(white)
#save the files to output folder with the name "Image_x"
filename = "output/Image_%s" %f
plt.colorbar()
# live image display
plt.draw()
# need to add pause command, otherwise it does not work
plt.pause(.01)
# clear the figure to avoid memory issues
plt.clf()
#save these contours (outputs) as bmps with same file names
cv2.imwrite(filename,white)
答案 0 :(得分:1)
似乎我们需要首先添加matcher2.py和matcher.py。否则,将出现以下错误。
>>> import os, glob
>>> os.system("tree ./matcher")
sh:1:树:未找到
32512
答案 1 :(得分:0)
我们做一个小实验。
glob.glob("matcher/*.py")
如您所见,filename = "output/Image_%s" %f
的结果包含根目录“matcher /”。这就是说,写filename = "output/" + f.split("/")[-1]
是错误的。将其更改为## source bmps in "input/"
## processed bmps to "output"
for f in glob.glob("input/*.bmp"):
pass
filename = "output/" + f.split("/")[-1]
cv2.imwrite(filename, xxx)
左右。
{{1}}