change to simpledataset

c0acf6bc · tink2123 · 96cd6da1 · c0acf6bc
Commit c0acf6bc authored Nov 24, 2021 by tink2123
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 4 deletions

test_tipc/configs/rec_mv3_none_bilstm_ctc_v2.0/rec_icdar15_train.yml ...onfigs/rec_mv3_none_bilstm_ctc_v2.0/rec_icdar15_train.yml +6 -4

No files found.
--- a/test_tipc/configs/rec_mv3_none_bilstm_ctc_v2.0/rec_icdar15_train.yml
+++ b/test_tipc/configs/rec_mv3_none_bilstm_ctc_v2.0/rec_icdar15_train.yml
@@ -58,8 +58,9 @@ Metric:

 Train:
  dataset:
-    name: LMDBDataSet
-    data_dir: ./train_data/data_lmdb_release/training/
+    name: SimpleDataSet
+    data_dir: ./train_data/ic15_data/
+    label_file_list: ["./train_data/ic15_data/rec_gt_train.txt"]
    transforms:
      - DecodeImage: # load image
          img_mode: BGR
@@ -77,8 +78,9 @@ Train:

 Eval:
  dataset:
-    name: LMDBDataSet
-    data_dir: ./train_data/data_lmdb_release/validation/
+    name: SimpleDataSet
+    data_dir: ./train_data/ic15_data
+    label_file_list: ["./train_data/ic15_data/rec_gt_test.txt"]
    transforms:
      - DecodeImage: # load image
          img_mode: BGR