mirror of
https://github.com/PaddlePaddle/PaddleOCR.git
synced 2025-08-12 10:37:27 +00:00
modify normalization of db_process to speed up
This commit is contained in:
parent
244a06d542
commit
3f824f00bc
@ -194,8 +194,12 @@ class DBProcessTest(object):
|
||||
img_std = [0.229, 0.224, 0.225]
|
||||
im = im.astype(np.float32, copy=False)
|
||||
im = im / 255
|
||||
im -= img_mean
|
||||
im /= img_std
|
||||
im[:, :, 0] -= img_mean[0]
|
||||
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)
|
||||
im = im.transpose(channel_swap)
|
||||
return im
|
||||
|
Loading…
x
Reference in New Issue
Block a user