From 2d0be7dcc66e2f397d202c8c4eb8c53ffa9ed891 Mon Sep 17 00:00:00 2001 From: dx-tan Date: Wed, 6 Dec 2023 15:52:06 +0700 Subject: [PATCH] Update: text detection and recognition weight --- cope2n-ai-fi/configs/sdsap_sbt/configs.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cope2n-ai-fi/configs/sdsap_sbt/configs.py b/cope2n-ai-fi/configs/sdsap_sbt/configs.py index a84c774..2bc5edb 100644 --- a/cope2n-ai-fi/configs/sdsap_sbt/configs.py +++ b/cope2n-ai-fi/configs/sdsap_sbt/configs.py @@ -1,12 +1,12 @@ device = "cuda:0" ocr_engine = { "detector": { - "version": "/workspace/cope2n-ai-fi/weights/models/ocr_engine/sdsvtd/epoch_100_params.pth", + "version": "/workspace/cope2n-ai-fi/weights/models/ocr_engine/sdsvtd/epoch_68_params.pth", "rotator_version": "/workspace/cope2n-ai-fi/weights/models/ocr_engine/sdsvtd/best_bbox_mAP_epoch_30_lite.pth", "device": device }, "recognizer": { - "version": "/workspace/cope2n-ai-fi/weights/models/sdsvtr/hub/jxqhbem4to.pth", + "version": "/workspace/cope2n-ai-fi/weights/models/sdsvtr/hub/satrn_lite_general_sbt_imei_20231205_params.pth", "device": device }, "deskew": {