diff --git a/ppocr/data/det/db_process.py b/ppocr/data/det/db_process.py index cdb8efc2bd9028ff5fc004f13b6505db5d74328c..b64b8c8d227f293aff0eff90d1d85dee8dd85fce 100644 --- a/ppocr/data/det/db_process.py +++ b/ppocr/data/det/db_process.py @@ -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