hanz245 commited on
Commit
2ebdbb7
·
verified ·
1 Parent(s): 966abb3

Update CRNN+CTC/inference.py

Browse files
Files changed (1) hide show
  1. 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 best_model_v4.pth
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/best_model.pth',
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