ReadingPipeline-Peter / ocr /ocr_config.json
Stanislav Kalinin
feat: Update OCR models
d9c5d39
{
"alphabet": " #()+/0123456789[]abdefghiklmnoprstuw|×ǂабвгдежзийклмнопрстуфхцчшщъыьэюяѣ–⊕⊗",
"save_dir": "/home/jovyan/skalinin/OCR-model-replace-layer/OCR-model/data/experiments_peter/change_ocr_layer_for_ONNX_2",
"num_epochs": 200,
"pretrain_path": "/home/jovyan/skalinin/OCR-model-replace-layer/OCR-model/data/peter_pretrain/ocr_model.ckpt",
"image": {
"width": 1024,
"height": 128
},
"train": {
"datasets": [
{
"csv_path": "/home/jovyan/data/ocr/peter_mask/train.csv",
"prob": 10
},
{
"csv_path": "/home/jovyan/skalinin/StackMIX-PETER-SYNTH/StackMix-OCR/generated_synth/stmix-peter-synth-mask-part1.csv",
"prob": 20
},
{
"csv_path": "/home/jovyan/skalinin/StackMIX-PETER-SYNTH/StackMix-OCR/generated_synth/stmix-peter-synth-mask-part2.csv",
"prob": 20
},
{
"csv_path": "/home/jovyan/skalinin/StackMIX-PETER-SYNTH/StackMix-OCR/generated_synth/stmix-peter-synth-part-4-fix.csv",
"prob": 10
},
{
"csv_path": "/home/jovyan/skalinin/StackMIX-PETER-SYNTH/StackMix-OCR/generated_synth/stmix-peter-synth-part-5-fix.csv",
"prob": 10
},
{
"csv_path": "/home/jovyan/skalinin/StackMIX-PETER-SYNTH/StackMix-OCR/peter_mask_crops/peter_mask/stackmix_crops_ocr_format.csv",
"prob": 10
}
],
"epoch_size": 40000,
"batch_size": 128
},
"val": {
"datasets": [
{
"csv_path": "/home/jovyan/data/ocr/peter_mask/val.csv",
"prob": 1
}
],
"epoch_size": null,
"batch_size": 128
},
"test": {
"datasets": [
{
"csv_path": "/home/jovyan/data/ocr/peter_mask/test.csv",
"prob": 1
}
],
"epoch_size": null,
"batch_size": 128
}
}