change to simpledataset

pull/4739/head
tink2123 2021-11-24 08:24:48 +00:00
parent 96cd6da1c9
commit c0acf6bc27
1 changed files with 6 additions and 4 deletions

View File

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