mirror of
https://github.com/qurator-spk/eynollah.git
synced 2026-06-16 09:59:13 +02:00
Merge branch 'integrating_trocr_and_torch_ensembling_and_updating_characters_list'
# Conflicts: # src/eynollah/eynollah.py # src/eynollah/eynollah_ocr.py # src/eynollah/patch_encoder.py # src/eynollah/training/cli.py # src/eynollah/training/gt_gen_utils.py # src/eynollah/training/inference.py # src/eynollah/training/models.py # src/eynollah/training/train.py # src/eynollah/training/utils.py # src/eynollah/training/weights_ensembling.py # train/requirements.txt
This commit is contained in:
commit
6df11d92d8
16 changed files with 1048 additions and 245 deletions
|
|
@ -1,17 +1,17 @@
|
|||
{
|
||||
"backbone_type" : "transformer",
|
||||
"task": "cnn-rnn-ocr",
|
||||
"task": "transformer-ocr",
|
||||
"n_classes" : 2,
|
||||
"max_len": 280,
|
||||
"n_epochs" : 3,
|
||||
"max_len": 192,
|
||||
"n_epochs" : 1,
|
||||
"input_height" : 32,
|
||||
"input_width" : 512,
|
||||
"weight_decay" : 1e-6,
|
||||
"n_batch" : 4,
|
||||
"n_batch" : 1,
|
||||
"learning_rate": 1e-5,
|
||||
"save_interval": 1500,
|
||||
"patches" : false,
|
||||
"pretraining" : true,
|
||||
"pretraining" : false,
|
||||
"augmentation" : true,
|
||||
"flip_aug" : false,
|
||||
"blur_aug" : true,
|
||||
|
|
@ -77,7 +77,6 @@
|
|||
"dir_output": "/home/vahid/extracted_lines/1919_bin/output",
|
||||
"dir_rgb_backgrounds": "/home/vahid/Documents/1_2_test_eynollah/set_rgb_background",
|
||||
"dir_rgb_foregrounds": "/home/vahid/Documents/1_2_test_eynollah/out_set_rgb_foreground",
|
||||
"dir_img_bin": "/home/vahid/extracted_lines/1919_bin/images_bin",
|
||||
"characters_txt_file":"/home/vahid/Downloads/models_eynollah/model_eynollah_ocr_cnnrnn_20250930/characters_org.txt"
|
||||
"dir_img_bin": "/home/vahid/extracted_lines/1919_bin/images_bin"
|
||||
|
||||
}
|
||||
|
|
|
|||
|
|
@ -8,3 +8,8 @@ tensorflow-addons # for connected_components, depublished and only compatible wi
|
|||
tensorflow < 2.16 # for tensorflow-addons, so only needed in training
|
||||
tf_data < 2.16 # for tensorflow-addons, so only needed in training
|
||||
protobuf < 5 # for tensorflow-addons, so only needed in training
|
||||
torch
|
||||
evaluate
|
||||
accelerate
|
||||
jiwer
|
||||
transformers <= 4.30.2
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue