提取文件夹下所有文件名 保存在txt文本 ，Python实现1

import os 
def ListFilesToTxt(dir,file,wildcard,recursion):
    exts = wildcard.split(" ")
    files = os.listdir(dir)
    for name in files:
        fullname=os.path.join(dir,name)
        if(os.path.isdir(fullname) & recursion):
            ListFilesToTxt(fullname,file,wildcard,recursion)
        else:
            for ext in exts:
                if(name.endswith(ext)):
                    file.write(name + "\n")
                    break
def Test():
  dir="spliced_NIST/rgb_imgs/"     #文件路径
  outfile="binaries.txt"                     #写入的txt文件名
  wildcard = ".png"      #要读取的文件类型；
  file = open(outfile,"w")
  if not file:
    print ("cannot open the file %s for writing" % outfile)
  ListFilesToTxt(dir,file,wildcard, 1)
  file.close()
Test()

2  提取npy文件保存为图片

import numpy as np
import cv2
import os 


dir="save_preds/voc2010_model.ckpt-