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

* Flexible ways of getting file name Address issue open-mmlab#1078 * fix lint Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
gaotongxiao · Jul 4, 2022 · 68bb3e6 · 68bb3e6
1 parent 20f2382
commit 68bb3e6
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,