def writeFiles(self, outDir, writeLabels=True): dir_tools.mkdir_ask(outDir) if writeLabels: ann.MultiLabeler.writeFiles(self, outDir) with open(os.path.join(outDir, "classifier.pkl"), "w") as fh: cPickle.dump(self.cls, fh) np.savetxt(os.path.join(outDir, "coeffs.txt"), self.cls.classifier.coef_) np.savetxt(os.path.join(outDir, "intercepts.txt"), self.cls.classifier.intercept_)
def writeFiles(self, outDir): dir_tools.mkdir_ask(outDir) for (i,image) in enumerate(self.images): outFile = "%s.label.png"%os.path.basename(self.imageFiles[i]) cv2.imwrite(os.path.join(outDir,outFile),self.labels[i]) with open(os.path.join(outDir,"label_info.json"),"w") as fh: json.dump( dict([(i, "LABEL_DESCRIPTION") for i in xrange(self.labelWindow.maxLabel)]), fh) print "wrote files to",outDir
def __init__(self, topic, extension): self.topic = topic self.extension = extension dir_tools.mkdir_ask(topicPath(topic), make_path=True)
def resetDataDir(): dir_tools.mkdir_ask(DATA_ROOT, make_path=True) os.mkdir(join(DATA_ROOT, "once"))
def __init__(self, topic, extension): Publisher.__init__(self, topic) self.names = Names(topic, extension) dir_tools.mkdir_ask(topicPath(topic), make_path=True)