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