def main_function(): parser = argparse.ArgumentParser( description='Test a trained neuronal network.', ) parser.add_argument('-i', '--image', action='store', dest='image', required=True, help='The path to the input image file.') parser.add_argument('-g', '--graph', action='store', dest='graph', required=True, help='The path to the graph file.') parser.add_argument('-l', '--label', action='store', dest='label', required=True, help='The path to the label file.') args = parser.parse_args() for key, value in vars(args).items(): check_file(value) detector = Detector(graph_file=args.graph, labels_file=args.label) image = Image.open(args.image) answer = detector.detect([image]) print(answer)
def load_convnet(self): self.convnet = Detector(labels_file=self.configuration.labels, graph_file=self.configuration.network) if not self.configuration.word in self.convnet.labels: error_message = self.configuration.word + " is not in label file." logger.error(error_message) raise Exception(error_message)
def detector(): return Detector()
def load_convnet(self): self.convnet = Detector() if not self.search.word in self.convnet.labels: error_message = self.search.word + " is not in label file." self.logger.error(error_message) raise Exception(error_message)