Unverified Commit 42d2f7af authored by dyning's avatar dyning Committed by GitHub
Browse files

Merge pull request #366 from BeyondYourself/develop

add the missing improt for east_process.py and format some codes
parents 455769b1 2f87ed81
......@@ -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