diff --git a/ppocr/data/det/east_process.py b/ppocr/data/det/east_process.py index 7f0deda2..f9a07440 100755 --- a/ppocr/data/det/east_process.py +++ b/ppocr/data/det/east_process.py @@ -16,7 +16,7 @@ import math import cv2 import numpy as np import json - +import sys class EASTProcessTrain(object): def __init__(self, params): @@ -78,7 +78,7 @@ class EASTProcessTrain(object): dst_polys = [] rand_degree_ratio = np.random.rand() rand_degree_cnt = 1 - if rand_degree_ratio > 0.333 and rand_degree_ratio < 0.666: + if 0.333 < rand_degree_ratio < 0.666: rand_degree_cnt = 2 elif rand_degree_ratio > 0.666: rand_degree_cnt = 3 @@ -138,7 +138,7 @@ class EASTProcessTrain(object): continue if p_area > 0: #'poly in wrong direction' - if tag == False: + if not tag: tag = True #reversed cases should be ignore poly = poly[(0, 3, 2, 1), :] validated_polys.append(poly)