base_dir=/path/to/dimt_ocr_free | |
donut_dir=/path/to/donut-base | |
nougat_dir=/path/to/nougat-small | |
checkpoint_dir=/path/to/checkpoint_dir | |
image_file_path=/path/to/image.png | |
export CUDA_VISIBLE_DEVICES=0 | |
python inference.py \ | |
--base_dir $base_dir \ | |
--donut_dir $donut_dir \ | |
--nougat_dir $nougat_dir \ | |
--checkpoint_dir $checkpoint_dir \ | |
--image_file_path $image_file_path |