diff --git a/mmocr/datasets/pipelines/textdet_targets/textsnake_targets.py b/mmocr/datasets/pipelines/textdet_targets/textsnake_targets.py index eeebf5bd..73f4597c 100644 --- a/mmocr/datasets/pipelines/textdet_targets/textsnake_targets.py +++ b/mmocr/datasets/pipelines/textdet_targets/textsnake_targets.py @@ -188,6 +188,7 @@ class TextSnakeTargets(BaseTextDetTargets): assert line.shape[0] >= 2 assert line.shape[1] == 2 assert isinstance(n, int) + assert n > 0 length_list = [ norm(line[i + 1] - line[i]) for i in range(len(line) - 1) @@ -234,8 +235,8 @@ class TextSnakeTargets(BaseTextDetTargets): resampled_line2 (ndarray): The resampled line 2. """ - assert sideline1.ndim == sideline1.ndim == 2 - assert sideline1.shape[1] == sideline1.shape[1] == 2 + assert sideline1.ndim == sideline2.ndim == 2 + assert sideline1.shape[1] == sideline2.shape[1] == 2 assert sideline1.shape[0] >= 2 assert sideline2.shape[0] >= 2 assert isinstance(resample_step, float) @@ -250,7 +251,7 @@ class TextSnakeTargets(BaseTextDetTargets): ]) total_length = (length1 + length2) / 2 - resample_point_num = int(float(total_length) / resample_step) + resample_point_num = max(int(float(total_length) / resample_step), 1) resampled_line1 = self.resample_line(sideline1, resample_point_num) resampled_line2 = self.resample_line(sideline2, resample_point_num)