move TF+Keras dependencies to training extra, replace by ONNX+TRT

This commit is contained in:
Robert Sachunsky 2026-07-07 22:56:36 +02:00
parent 1567df1379
commit 171a8a3161
2 changed files with 4 additions and 3 deletions

View file

@ -1,7 +1,7 @@
# ocrd includes opencv, numpy, shapely, click
ocrd >= 3.3.0
onnxruntime-gpu[cuda,cudnn] # w/ .onnx models
tensorrt_cu12 < 11 # 11 incompatible with CUDA libs from onnxruntime-gpu[cuda,cudnn]
scikit-learn >= 0.23.2
tensorflow
tf-keras # avoid keras 3 (also needs TF_USE_LEGACY_KERAS=1)
scikit-image
tabulate