qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-07-11 22:29:29 +02:00
Code Issues Projects Releases Wiki Activity

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

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

4
requirements.txt
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

3
train/requirements.txt
View file

@ -6,5 +6,6 @@ imutils
scipy
tensorflow-addons # for connected_components, depublished and only compatible with tensorflow < 2.16
tensorflow < 2.16 # for tensorflow-addons, so only needed in training
tf_data < 2.16 # for tensorflow-addons, so only needed in training
tf-keras # avoid keras 3 (also needs TF_USE_LEGACY_KERAS=1)
tf-data < 2.16 # for tensorflow-addons, so only needed in training
protobuf < 5 # for tensorflow-addons, so only needed in training