def main():
myImager = Imager()
#myModel = Model()
while True:
img = myImager.process_img(myImager.capture_img())
if not img is None:
myImager.show_img(img)
import json
from imager import Imager
from os import listdir, getcwd, walk
from os.path import isfile, join
inDir = input('Directories -\n{}\nInput directory: '.format(
next(walk('.'))[1]))
myImager = Imager()
dataset = []
print('Processing data')
for picFile in [
fileName if fileName.endswith('.jpg') else ''
for fileName in listdir(getcwd() + '/' + inDir)
]:
rawImg = myImager.open_img(inDir + '/' + picFile)
procdImg = myImager.process_img(rawImg)
if procdImg is None: continue
#myImager.show_img(procdImg)
dataset.append([procdImg.tolist(),
int(picFile.split()[0].split('.')[0])
]) #picFile.split()[0]])
print('Writing to dataset file')
with open(inDir + '-dataset.json', 'w') as dsFile:
dsFile.write(json.dumps(dataset))
print('Sucessfully wrote data set to JSON file')
