"magic_pdf/vscode:/vscode.git/clone" did not exist on "74e954daceae231f6775723f7f24538a65171c17"
Unverified Commit 2f87ed81 authored by shaohua.zhang's avatar shaohua.zhang Committed by GitHub
Browse files

add the missing import and format some codes

add the missing import and format some codes
parent 26219d5f
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment