ReadingPipeline-Peter / ocr /ocr_config.json
Stanislav Kalinin
feat: Add weights and configs for Peter models
54c5290
raw
history blame
No virus
2.03 kB
{
"alphabet": " #()+/0123456789[]abdefghiklmnoprstuw|×ǂабвгдежзийклмнопрстуфхцчшщъыьэюяѣ–⊕⊗",
"save_dir": "/home/jovyan/skalinin/OCR-model/data/experiments_peter/peter_mask_and_stmix_synth_bbox-4",
"num_epochs": 100,
"pretrain_path": "",
"image": {
"width": 1024,
"height": 128
},
"train": {
"datasets": [
{
"csv_path": "/home/jovyan/data/ocr/peter_mask/train.csv",
"prob": 15
},
{
"csv_path": "/home/jovyan/skalinin/StackMIX-PETER-SYNTH/StackMix-OCR/generated_synth/stmix-peter-synth-part-3.csv",
"prob": 5
},
{
"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/generated_synth/stmix-peter-synth-mask-part1.csv",
"prob": 30
},
{
"csv_path": "/home/jovyan/skalinin/StackMIX-PETER-SYNTH/StackMix-OCR/peter_mask_crops/peter_mask/stackmix_crops_ocr_format.csv",
"prob": 20
}
],
"epoch_size": 40000,
"batch_size": 64
},
"val": {
"datasets": [
{
"csv_path": "/home/jovyan/data/ocr/peter_mask/val.csv",
"prob": 1
}
],
"epoch_size": null,
"batch_size": 64
},
"test": {
"datasets": [
{
"csv_path": "/home/jovyan/data/ocr/peter_mask/test.csv",
"prob": 1
}
],
"epoch_size": null,
"batch_size": 64
}
}