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()