====== batch generator.py ====== import os featuresbat = open("extractfeatures.bat", 'w') matchesbat = open("computematches.bat", 'w') matchesfile = open("matlabinputfile.txt",'w') file = 0 image = 0 # How many frames do we skip between the images? frames_to_skip = 15 if (os.path.isdir('images')!= 1): print("Image file directory not found") exit() if (os.path.isdir('FEATURES')!= 1): os.mkdir('FEATURES') if (os.path.isdir('MATCHES')!= 1): os.mkdir('MATCHES') for dirpath, dirnames, files in os.walk('images'): for filename in files: featuresbat.write("\"EXE/features.exe\" 0 images/%s.pgm FEATURES/%s.key\n" %(filename[:-4], filename[:-4])) if (file > 0): if (image + frames_to_skip > 99): if (image + frames_to_skip > 999): matchesbat.write("\"EXE/match.exe\" FEATURES/%s.key FEATURES/%s%s.key MATCHES/frames_%s-%s.txt\n" %(oldfile[:-4], filename[:-8], image+frames_to_skip, image, image+frames_to_skip)) else: matchesbat.write("\"EXE/match.exe\" FEATURES/%s.key FEATURES/%s%s.key MATCHES/frames_%s-%s.txt\n" %(oldfile[:-4], filename[:-7], image+frames_to_skip, image, image+frames_to_skip)) elif (image + frames_to_skip < 99): matchesbat.write("\"EXE/match.exe\" FEATURES/%s.key FEATURES/%s0%s.key MATCHES/frames_%s-%s.txt\n" %(oldfile[:-4], filename[:-7], image+frames_to_skip, image, image+frames_to_skip)) matchesfile.write("frames_%s-%s.txt\n" %(image, image+frames_to_skip)) file = file + 1 image = image + 1 oldfile = filename featuresbat.close() matchesbat.close() matchesfile.close()