Upload anytext.py
Browse files- anytext.py +1 -1
anytext.py
CHANGED
@@ -476,7 +476,7 @@ class TextEmbeddingModule(nn.Module):
|
|
476 |
args["rec_image_shape"] = "3, 48, 320"
|
477 |
args["rec_batch_num"] = 6
|
478 |
args["rec_char_dict_path"] = "./text_embedding_module/OCR/ppocr_keys_v1.txt"
|
479 |
-
args["use_fp16"] =
|
480 |
self.embedding_manager.recog = TextRecognizer(args, self.text_predictor)
|
481 |
|
482 |
@torch.no_grad()
|
|
|
476 |
args["rec_image_shape"] = "3, 48, 320"
|
477 |
args["rec_batch_num"] = 6
|
478 |
args["rec_char_dict_path"] = "./text_embedding_module/OCR/ppocr_keys_v1.txt"
|
479 |
+
args["use_fp16"] = use_fp16
|
480 |
self.embedding_manager.recog = TextRecognizer(args, self.text_predictor)
|
481 |
|
482 |
@torch.no_grad()
|