diff --git a/tools/infer_rec.py b/tools/infer_rec.py index 2c9b5f797a7404ac18b91e444dadf5be0d2765f6..b0c836ff3bcee5b9009b62c60483697db8091bde 100755 --- a/tools/infer_rec.py +++ b/tools/infer_rec.py @@ -137,7 +137,7 @@ def main(): if info is not None: logger.info("\t result: {}".format(info)) - fout.write(file + "\t" + info + "\n") + fout.write(os.path.basename(file) + "\t" + info + "\n") logger.info("success!")