fix conflicts

4 years ago · a5b219127f
parent 7cce85cc5c
commit a5b219127f
3 changed files with 2 additions and 2 deletions
--- a/configs/det/ch_ppocr_v1.1/ch_det_mv3_db_v2.0.yml
+++ b/configs/det/ch_ppocr_v1.1/ch_det_mv3_db_v2.0.yml
--- a/configs/det/ch_ppocr_v1.1/ch_det_res18_db_v2.0.yml
+++ b/configs/det/ch_ppocr_v1.1/ch_det_res18_db_v2.0.yml
--- 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,