개정판 33ccef35
issue #1366: log change
Change-Id: I898e041e1f92b6c76d8fc9dc90450ecc0333359b
DTI_PID/DTI_PID/TextDetector.py | ||
---|---|---|
317 | 317 |
if rect in rects: |
318 | 318 |
rects.remove(rect) |
319 | 319 |
else: |
320 |
print(str(rect)) |
|
320 |
pass |
|
321 |
#print(str(rect)) |
|
321 | 322 |
|
322 | 323 |
for merge in overlap_merges: |
323 | 324 |
max_x, max_y, min_x, min_y = 0, 0, sys.maxsize, sys.maxsize |
... | ... | |
416 | 417 |
if rect in rects: |
417 | 418 |
rects.remove(rect) |
418 | 419 |
else: |
419 |
print(str(rect)) |
|
420 |
pass |
|
421 |
#print(str(rect)) |
|
420 | 422 |
|
421 | 423 |
for merge in v_merges + h_merges: |
422 | 424 |
max_x, max_y, min_x, min_y = 0, 0, sys.maxsize, sys.maxsize |
DTI_PID/WebServer/CRAFT_pytorch_master/text_craft.py | ||
---|---|---|
246 | 246 |
args.poly = True |
247 | 247 |
|
248 | 248 |
box_craft_list = [] |
249 |
t = time.time() |
|
249 | 250 |
for img in imgs: |
250 | 251 |
if img.shape[0] == 2: img = img[0] |
251 | 252 |
if len(img.shape) == 2 : img = cv2.cvtColor(img, cv2.COLOR_GRAY2RGB) |
252 | 253 |
if img.shape[2] == 4: img = img[:,:,:3] |
253 | 254 |
image = np.array(img) |
254 | 255 |
|
255 |
t = time.time() |
|
256 |
|
|
257 | 256 |
bboxes, polys, score_text = test_net(net, image, args.text_threshold, args.link_threshold, args.low_text, args.cuda, args.poly, refine_net, args) |
258 | 257 |
|
259 | 258 |
# save score text |
... | ... | |
278 | 277 |
if img_path: |
279 | 278 |
cv2.imwrite(img_path, img) |
280 | 279 |
|
281 |
print("elapsed time : {}s".format(time.time() - t)) |
|
282 |
|
|
283 | 280 |
box_craft_list.append(box_craft) |
281 |
|
|
282 |
print("total elapsed time : {}s, tile number : {}, sec per tile : {}".format(time.time() - t, len(imgs), (time.time() - t) / len(imgs))) |
|
284 | 283 |
|
285 | 284 |
return [box_craft_list] |
286 | 285 |
|
DTI_PID/WebServer/app.py | ||
---|---|---|
78 | 78 |
return jsonify({'text_box_list': boxes_list}) |
79 | 79 |
|
80 | 80 |
if __name__ == '__main__': |
81 |
app.run(debug=True) |
|
81 |
app.run(debug=False) |
내보내기 Unified diff