EPECscan / requirements.txt
chk03042010's picture
Update requirements.txt
b5de14a verified
# Ultralytics requirements
# Usage: pip install -r requirements.txt
# Base ----------------------------------------
hydra-core
matplotlib
numpy
opencv-python
Pillow
PyYAML
requests
scipy
yolo-cam
torch
torchvision
tqdm
ultralytics
grad-cam
tensorflow
Cython
# Logging -------------------------------------
tensorboard
# clearml
# comet
#tracking
filterpy
scikit-image
# Plotting ------------------------------------
pandas
seaborn
# Export --------------------------------------
# coremltools # CoreML export
# onnx # ONNX export
# onnx-simplifier # ONNX simplifier
# nvidia-pyindex # TensorRT export
# nvidia-tensorrt # TensorRT export
# scikit-learn # CoreML quantization
# tensorflow> # TF exports (-cpu, -aarch64, -macos)
# tensorflowjs # TF.js export
# openvino-dev # OpenVINO export
# Extras --------------------------------------
ipython # interactive notebook
psutil # system utilization
thop # FLOPs computation
# albumentations
# pycocotools # COCO mAP
# roboflow
# HUB -----------------------------------------
GitPython