工作学习第二--试错总结与学习进步（2）

补充前面的内容：

文件操作：把一个文件夹中的图片名写入txt文件：

import os
def ListFilesToTxt(dir,file,wildcard,recursion):
    exts = wildcard.split(" ")
    for root, subdirs, files in os.walk(dir):
        for name in files:
            for ext in exts:
                if(name.endswith(ext)):
                    file.write(name + "\n")
                    break
        if(not recursion):
            break
def Test():
  dir="E:\\python\\image_diff\\test2\\short_circuit_test"
  outfile="get_name_test.txt"
  wildcard = ".bmp"

  file = open(outfile,"w")
  if not file:
    print ("cannot open the file %s for writing" % outfile)
  ListFilesToTxt(dir,file,wildcard, 0)

  file.close()
Test()

批量读取txt文件中的图片：

fopen = open("E:\\python\\image_diff\\test2\\get_name_2.txt",'r')
lines = fopen.readlines()
imgname = []
for line1 in lines:
	line = line1.replace('\n','')
	imgname.append(line)
for i in range(0, len(imgname), 2):
	# Read the images to be aligned
	im1 = cv2.imread("./short_circuit_2/"+imgname[i])
	im2 = cv2.imread("./short_circuit_2/"+imgname[i+1])

 

在python-pycharm中程序运行完不会自动退出运行：

import sys          sys.exit()   即可