|
|
|
@ -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)
|
|
|
|
|