required=True, help='input an image to predict') parser.add_argument('--model', default='vgg-22class-200epoch-20170609053802', help='trained model in 90_result') return parser.parse_args() if __name__ == '__main__': try: logger.info("------ start ------") utils.lock() args = parse_args() if not os.path.exists(args.image): raise LunaExcepion(config.inputerr) from fcn import Fcn fcn = Fcn() fcn.load(args.model) logger.info('model loaded.') img, labels = fcn.predict(args.image) logger.info('predict end.') print({"error": "", "data": {"img": img, "labels": labels}}) except (KeyboardInterrupt, SystemExit): utils.unlock() utils.error(config.syserr) except LunaExcepion as e:
def lock(): if os.path.exists(config.lock_file): raise LunaExcepion(config.locked) lock_file = open(config.lock_file, 'w') lock_file.write(str(os.getpid())) lock_file.close()