mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

Add TensorRT dependencies (#8553)

Browse Source 
Update requirements.txt
This commit is contained in:
曾逸夫（Zeng Yifu） 2022-07-12 19:19:25 +08:00 committed by GitHub
parent 04146371b9
commit fbd3020525
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
requirements.txt
View File

@ -26,6 +26,8 @@ seaborn>=0.11.0
# coremltools>=4.1 # CoreML export # coremltools>=4.1 # CoreML export
# onnx>=1.9.0 # ONNX export # onnx>=1.9.0 # ONNX export
# onnx-simplifier>=0.3.6 # ONNX simplifier # onnx-simplifier>=0.3.6 # ONNX simplifier
# nvidia-pyindex # TensorRT export
# nvidia-tensorrt # TensorRT export
# scikit-learn==0.19.2 # CoreML quantization # scikit-learn==0.19.2 # CoreML quantization
# tensorflow>=2.4.1 # TFLite export # tensorflow>=2.4.1 # TFLite export
# tensorflowjs>=3.9.0 # TF.js export # tensorflowjs>=3.9.0 # TF.js export