forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

modify normalization of db_process to speed up

This commit is contained in:
LDOUBLEV 2020-07-10 10:40:10 +08:00
parent 244a06d542
commit 3f824f00bc
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ppocr/data/det/db_process.py
View File

@ -194,8 +194,12 @@ class DBProcessTest(object):
img_std = [0.229, 0.224, 0.225] img_std = [0.229, 0.224, 0.225]
im = im.astype(np.float32, copy=False) im = im.astype(np.float32, copy=False)
im = im / 255 im = im / 255
im -= img_mean im[:, :, 0] -= img_mean[0]
im /= img_std im[:, :, 1] -= img_mean[1]
im[:, :, 2] -= img_mean[2]
im[:, :, 0] /= img_std[0]
im[:, :, 1] /= img_std[1]
im[:, :, 2] /= img_std[2]
channel_swap = (2, 0, 1) channel_swap = (2, 0, 1)
im = im.transpose(channel_swap) im = im.transpose(channel_swap)
return im return im