[Fix] Flexible ways of getting file name (#1107)

* Flexible ways of getting file name Address issue #1078 * fix lint Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
open-mmlab · Jul 4, 2022 · 7800e13 · 7800e13
1 parent 64fb6ff
commit 7800e13
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/data/textdet/lv_converter.py b/tools/data/textdet/lv_converter.py
@@ -83,7 +83,7 @@ def load_img_info(files):
         '.')[0]
     # read imgs while ignoring orientations
     img = mmcv.imread(img_file, 'unchanged')
-    img_file = img_file.split('data/lv/')[1]
+    img_file = os.path.split(img_file)[-1]
 
     img_info = dict(
         file_name=img_file,