diff --git a/configs/det/ch_ppocr_v1.1/ch_det_mv3_db.yml b/configs/det/ch_ppocr_v1.1/ch_det_mv3_db_v2.0.yml similarity index 100% rename from configs/det/ch_ppocr_v1.1/ch_det_mv3_db.yml rename to configs/det/ch_ppocr_v1.1/ch_det_mv3_db_v2.0.yml diff --git a/configs/det/ch_ppocr_v1.1/ch_det_res18_db.yml b/configs/det/ch_ppocr_v1.1/ch_det_res18_db_v2.0.yml similarity index 100% rename from configs/det/ch_ppocr_v1.1/ch_det_res18_db.yml rename to configs/det/ch_ppocr_v1.1/ch_det_res18_db_v2.0.yml diff --git a/ppocr/data/simple_dataset.py b/ppocr/data/simple_dataset.py index 168913265b069c74ce23774699ff38f0d29a8589..d2069d0d5fb608650bbaa2002a27a59265e334c5 100644 --- a/ppocr/data/simple_dataset.py +++ b/ppocr/data/simple_dataset.py @@ -57,9 +57,9 @@ class SimpleDataSet(Dataset): sample_num = round(len(datas) * sample_ratio) if data_num_per_epoch is not None: - sample_num = data_num_per_epoch * sample_ratio + sample_num = int(data_num_per_epoch * sample_ratio) - nums, rem = sample_num // len(datas), sample_num % len(datas) + nums, rem = int(sample_num // len(datas)), int(sample_num % len(datas)) return list(datas) * nums + random.sample(datas, rem) def get_image_info_list(self,