Update requirements.txt (#9)
parent
39e9f3cae9
commit
745f8e3759
@ -0,0 +1,46 @@
|
||||
# Ultralytics requirements
|
||||
# Usage: pip install -r requirements.txt
|
||||
|
||||
# Base ----------------------------------------
|
||||
matplotlib>=3.2.2
|
||||
numpy>=1.18.5
|
||||
opencv-python>=4.1.1
|
||||
Pillow>=7.1.2
|
||||
PyYAML>=5.3.1
|
||||
requests>=2.23.0
|
||||
scipy>=1.4.1
|
||||
torch>=1.7.0
|
||||
torchvision>=0.8.1
|
||||
tqdm>=4.64.0
|
||||
|
||||
# Logging -------------------------------------
|
||||
tensorboard>=2.4.1
|
||||
# clearml
|
||||
# comet
|
||||
|
||||
# Plotting ------------------------------------
|
||||
pandas>=1.1.4
|
||||
seaborn>=0.11.0
|
||||
|
||||
# Export --------------------------------------
|
||||
# coremltools>=5.2 # CoreML export
|
||||
# onnx>=1.9.0 # ONNX export
|
||||
# onnx-simplifier>=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 export (or tensorflow-cpu, tensorflow-aarch64)
|
||||
# tensorflowjs>=3.9.0 # TF.js export
|
||||
# openvino-dev # OpenVINO export
|
||||
|
||||
# Extras --------------------------------------
|
||||
ipython # interactive notebook
|
||||
psutil # system utilization
|
||||
thop>=0.1.1 # FLOPs computation
|
||||
# albumentations>=1.0.3
|
||||
# pycocotools>=2.0 # COCO mAP
|
||||
# roboflow
|
||||
|
||||
# HUB -----------------------------------------
|
||||
GitPython>=3.1.24
|
||||
requests
|
Loading…
Reference in new issue