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)
