从一个文件夹中提取图像,处理它并使用python,opencv保存到另一个文件夹中

时间:2017-10-25 14:10:16

标签: python opencv save image-extraction

我正在尝试从一个子文件夹“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)

2 个答案:

答案 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}}