class Engine:
def __init__(self, preloaded = True, pickleName = "PythonPackage/svmResults.pkl"):
self.vm = VectorMachine()
if preloaded:
self.svm = self.vm.readFromPickle(pickleName)
else:
self.svm = None
print("Idk man you're f****d")
def isWriteDone(self, filename):
#checking if the file is still being written by cam app
file_stat = os.stat(filename);
prev_size = file_stat.st_size;
time.sleep(.5)
file_stat = os.stat(filename);
size = file_stat.st_size;
return size == prev_size
def runEngine(self, fromdir="prep_dir/", todir="trash_dir/", ext=".JPG"):
while True:
for f in os.listdir(fromdir):
if not f.endswith(ext) and not f.endswith(ext.lower()):
print("Files not correct format")
continue
try:
while not self.isWriteDone(fromdir+f):
print ("not done writing")
#wait until file is done
print(self.svm.predict([self.vm.getCompressedRaveledPic(fromdir, os.path.splitext(f)[0], ext, "", "")]))
shutil.move(fromdir+f, todir+f)
except OSError:
print("Possible file corrupted : " + str(f) + ". We will proceed to remove it.")
os.remove(fromdir+f)
def __init__(self, preloaded = True, pickleName = "PythonPackage/svmResults.pkl"):
self.vm = VectorMachine()
if preloaded:
self.svm = self.vm.readFromPickle(pickleName)
else:
self.svm = None
print("Idk man you're f****d")
