33 lines
1.4 KiB
Bash
Executable file
33 lines
1.4 KiB
Bash
Executable file
#!/usr/bin/env bash
|
|
# Runner for image2text service on Jetson Nano
|
|
# Adjust env vars below as needed.
|
|
|
|
set -euo pipefail
|
|
|
|
# ---- Configurable env ----
|
|
export MODEL_EDGETPU_PATH=${MODEL_EDGETPU_PATH:-./models/ssd_mobilenet_v2_coco_quant_postprocess_edgetpu.tflite}
|
|
export MODEL_LABELS_PATH=${MODEL_LABELS_PATH:-./models/coco_labels.txt}
|
|
export MODEL_EAST_PATH=${MODEL_EAST_PATH:-./models/frozen_east_text_detection.pb}
|
|
export OCR_BACKEND=${OCR_BACKEND:-tesseract} # tesseract|paddle|auto
|
|
export OCR_LANG=${OCR_LANG:-eng+rus}
|
|
export OCR_AUTO_LANG=${OCR_AUTO_LANG:-true}
|
|
export OCR_CONFIG=${OCR_CONFIG:---oem 1 --psm 6}
|
|
export PADDLE_LANG=${PADDLE_LANG:-ru}
|
|
export PADDLE_USE_GPU=${PADDLE_USE_GPU:-false}
|
|
export PADDLE_USE_ANGLE_CLS=${PADDLE_USE_ANGLE_CLS:-false}
|
|
export PADDLE_DET_MODEL_DIR=${PADDLE_DET_MODEL_DIR:-./models/paddle/PP-OCRv3_mobile_det_infer}
|
|
export PADDLE_REC_MODEL_DIR=${PADDLE_REC_MODEL_DIR:-./models/paddle/cyrillic_PP-OCRv3_mobile_rec_infer}
|
|
export PADDLE_CLS_MODEL_DIR=${PADDLE_CLS_MODEL_DIR:-}
|
|
export PADDLE_CPU_THREADS=${PADDLE_CPU_THREADS:-4}
|
|
# Point to tessdata_best if installed
|
|
export TESSDATA_PREFIX=${TESSDATA_PREFIX:-}
|
|
export MAX_IMAGE_MB=${MAX_IMAGE_MB:-10}
|
|
export CONF_THRESHOLD=${CONF_THRESHOLD:-0.4}
|
|
export RETURN_BOXES=${RETURN_BOXES:-true}
|
|
# Enable CUDA DNN for EAST if your OpenCV is built with CUDA
|
|
export USE_CUDA_DNN=${USE_CUDA_DNN:-false}
|
|
|
|
HOST=${HOST:-0.0.0.0}
|
|
PORT=${PORT:-8000}
|
|
|
|
python3 app.py --host "$HOST" --port "$PORT"
|