def generateImages(rootPath, num=40, size=100, filterSize=15, chunkSize=10, sd=50, t="pos"): for i in range(num): if t == "pos": randomImage = generatePositionImage(size, chunkSize, filterSize, sd) else: randomImage = generateImage(size, chunkSize, filterSize, sd) path = join(rootPath, str(i) + ".png") # save the image saveMatrixAsImage(randomImage, path) # now that everything is saved in the filesystem, load it to the db saved = False base = basename(rootPath) name = base number = 0 while not saved: try: saved = True db.add_image_set(rootPath, name) except ValueError: name = base + str(number) number += 1 saved = False except TypeError: saved = False raise
def generateBW(rootPath, num=40, size=100): for i in range(num): randomImage = bwImage(size) path = join(rootPath, str(i) + ".png") # save the image saveMatrixAsImage(randomImage, path) # now that everything is saved in the filesystem, load it to the db saved = False base = basename(rootPath) name = base number = 0 while not saved: try: saved = True db.add_image_set(rootPath, name) except ValueError: name = base + str(number) number += 1 saved = False except TypeError: saved = False raise
import argparse import db parser = argparse.ArgumentParser(description='Upload a directory of image files as an image set') parser.add_argument('dir', help='The directory to upload') if __name__ == '__main__': args = parser.parse_args() db.add_image_set(args.dir)