mirror of
https://github.com/qurator-spk/eynollah.git
synced 2026-07-12 06:39:31 +02:00
move TF+Keras dependencies to training extra, replace by ONNX+TRT
This commit is contained in:
parent
1567df1379
commit
171a8a3161
2 changed files with 4 additions and 3 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue