From 1e527e77be8c89ae2e45e17e77be36e4088e8016 Mon Sep 17 00:00:00 2001 From: lizz Date: Tue, 25 May 2021 10:23:24 +0800 Subject: [PATCH] Make dirs for list_to_file (#235) Signed-off-by: lizz --- mmocr/utils/fileio.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/mmocr/utils/fileio.py b/mmocr/utils/fileio.py index 8dfd20c0..2157a1d4 100644 --- a/mmocr/utils/fileio.py +++ b/mmocr/utils/fileio.py @@ -1,3 +1,8 @@ +import os + +import mmcv + + def list_to_file(filename, lines): """Write a list of strings to a text file. @@ -5,6 +10,7 @@ def list_to_file(filename, lines): filename (str): The output filename. It will be created/overwritten. lines (list(str)): Data to be written. """ + mmcv.mkdir_or_exist(os.path.dirname(filename)) with open(filename, 'w', encoding='utf-8') as fw: for line in lines: fw.write(f'{line}\n')