get basename of filename

get basename of filename

get basename of filename
beae94c9 · livingbody · GitHub · 6bb09b03 · beae94c9
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

tools/infer_rec.py tools/infer_rec.py +1 -1

未找到文件。
--- 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!")