Stanislav Kalinin commited on
Commit
db56445
β€’
1 Parent(s): 6ffb2a1

chore: Rename model weights

Browse files
ocr/{corpus_o6.arpa β†’ kenlm_corpus.arpa} RENAMED
File without changes
ocr/{model-97-0.8179.ckpt β†’ ocr_model.ckpt} RENAMED
File without changes
pipeline_config.json CHANGED
@@ -3,7 +3,7 @@
3
  "input_format": "BGR"
4
  },
5
  "SegmPrediction": {
6
- "model_path": "segm/model-319-0.1578.ckpt",
7
  "config_path": "segm/segm_config.json",
8
  "device": "cuda"
9
  },
@@ -12,9 +12,9 @@
12
  },
13
  "ClassContourPosptrocess": {},
14
  "OCRPrediction": {
15
- "model_path": "ocr/model-97-0.8179.ckpt",
16
  "config_path": "ocr/ocr_config.json",
17
- "lm_path": "ocr/corpus_o6.arpa",
18
  "classes_to_ocr": ["shrinked_pupil_text", "shrinked_comments"],
19
  "device": "cuda"
20
  },
 
3
  "input_format": "BGR"
4
  },
5
  "SegmPrediction": {
6
+ "model_path": "segm/segm_model.ckpt",
7
  "config_path": "segm/segm_config.json",
8
  "device": "cuda"
9
  },
 
12
  },
13
  "ClassContourPosptrocess": {},
14
  "OCRPrediction": {
15
+ "model_path": "ocr/ocr_model.ckpt",
16
  "config_path": "ocr/ocr_config.json",
17
+ "lm_path": "ocr/kenlm_corpus.arpa",
18
  "classes_to_ocr": ["shrinked_pupil_text", "shrinked_comments"],
19
  "device": "cuda"
20
  },
segm/{model-319-0.1578.ckpt β†’ segm_model.ckpt} RENAMED
File without changes