-
Notifications
You must be signed in to change notification settings - Fork 0
/
requirements.txt
39 lines (38 loc) · 1.41 KB
/
requirements.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
#
# pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
# pip3 install "C:\Program Files\NVIDIA GPU Computing Toolkit\TensorRT\v8.6.1.6\graphsurgeon\graphsurgeon-0.4.6-py2.py3-none-any.whl"
# pip3 install "C:\Program Files\NVIDIA GPU Computing Toolkit\TensorRT\v8.6.1.6\uff\uff-0.6.9-py2.py3-none-any.whl"
# pip3 install "C:\Program Files\NVIDIA GPU Computing Toolkit\TensorRT\v8.6.1.6\onnx_graphsurgeon\onnx_graphsurgeon-0.3.12-py2.py3-none-any.whl"
# pip3 install "C:\Program Files\NVIDIA GPU Computing Toolkit\TensorRT\v8.6.1.6\python\tensorrt-8.6.1-cp311-none-win_amd64.whl"
#
numpy~=1.26.3
matplotlib~=3.8.1
scikit-image~=0.22.0
filterpy~=1.4.5
opencv-python~=4.10.0.82
ultralytics>=8.0.211
Flask~=3.0.3
Flask-HTTPAuth~=4.8.0
Flask-SocketIO~=5.3.6
Werkzeug~=3.0.3
imutils~=0.5.4
pyodbc~=5.1
mysql-connector-python~=9.0
MarkupSafe~=2.1.3
shapely~=2.0.2
torch>=2.2.2+cu118,<=3.2.1+cu118
psutil~=6.0.0
# For debug -----------------------------------
psutil~=6.0.0
gpustat~=1.1.1
# Export --------------------------------------
# coremltools>=7.0 # CoreML export
onnx~=1.16.0 # ONNX export
onnxsim>=0.4.1 # ONNX simplifier
# nvidia-pyindex # TensorRT export
# nvidia-tensorrt # TensorRT export
# scikit-learn==0.19.2 # CoreML quantization
# tensorflow>=2.4.1 # TF exports (-cpu, -aarch64, -macos)
# tflite-support
# tensorflowjs>=3.9.0 # TF.js export
# openvino-dev>=2023.0 # OpenVINO export