Update CRNN+CTC/inference.py
Browse files- CRNN+CTC/inference.py +2 -2
CRNN+CTC/inference.py
CHANGED
|
@@ -214,7 +214,7 @@ class CivilRegistryOCR:
|
|
| 214 |
def __init__(self, checkpoint_path, device='cuda', mode='auto', verbose=False):
|
| 215 |
"""
|
| 216 |
Args:
|
| 217 |
-
checkpoint_path : path to
|
| 218 |
device : 'cuda' or 'cpu'
|
| 219 |
mode : 'auto' → auto-detect per image (recommended)
|
| 220 |
'simple' → always use simple pipeline
|
|
@@ -340,7 +340,7 @@ def demo_inference():
|
|
| 340 |
print("=" * 70)
|
| 341 |
|
| 342 |
ocr = CivilRegistryOCR(
|
| 343 |
-
checkpoint_path='checkpoints/
|
| 344 |
device='cuda',
|
| 345 |
mode='auto',
|
| 346 |
verbose=True # shows which mode each image triggers
|
|
|
|
| 214 |
def __init__(self, checkpoint_path, device='cuda', mode='auto', verbose=False):
|
| 215 |
"""
|
| 216 |
Args:
|
| 217 |
+
checkpoint_path : path to best_model_v6.pth
|
| 218 |
device : 'cuda' or 'cpu'
|
| 219 |
mode : 'auto' → auto-detect per image (recommended)
|
| 220 |
'simple' → always use simple pipeline
|
|
|
|
| 340 |
print("=" * 70)
|
| 341 |
|
| 342 |
ocr = CivilRegistryOCR(
|
| 343 |
+
checkpoint_path='checkpoints/best_model_v6.pth',
|
| 344 |
device='cuda',
|
| 345 |
mode='auto',
|
| 346 |
verbose=True # shows which mode each image triggers
|