mirror of
https://github.com/RapidAI/RapidOCR.git
synced 2026-11-10 21:10:55 +08:00
perf(det): keep DetPreProcess mean/std in float32 (#756)
Signed-off-by: ExtReMLapin <3909752+ExtReMLapin@users.noreply.github.com>
This commit is contained in:
1 parent
b40c0f84b8
commit
558f566069
1 file changed
+2
-2
@@ -50,8 +50,8 @@ class DetPreProcess:
|
||||
if std is None:
|
||||
std = [0.5, 0.5, 0.5]
|
||||
|
||||
self.mean = np.array(mean)
|
||||
self.std = np.array(std)
|
||||
self.mean = np.array(mean, dtype=np.float32)
|
||||
self.std = np.array(std, dtype=np.float32)
|
||||
self.scale = 1 / 255.0
|
||||
|
||||
self.limit_side_len = limit_side_len
|
||||
|
||||
Reference in new issue
Block a user