Update requirements.txt (#9)
parent
39e9f3cae9
commit
745f8e3759
1 changed files with 46 additions and 0 deletions
@ -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