requirements.txt

# pip install -r requirements.txt

# base ----------------------------------------
dataclasses
requests>=2.23.0
Pillow>=7.1.2
PyYAML>=5.3.1
tqdm>=4.41.0
numpy>=1.18.5
matplotlib>=3.2.2
scipy>=1.4.1
scikit-image
scikit-learn==0.19.2
pandas>=1.1.4

# logging -------------------------------------
tensorboard>=2.4.1
wandb

# plotting ------------------------------------
seaborn>=0.11.0

# deep_sort -----------------------------------
easydict

# torchreid
h5py
six
tb-nightly
future
yacs
gdown
flake8
yapf
isort==4.3.21
imageio

# export --------------------------------------
# coremltools>=4.1  # CoreML export
onnx>=1.9.0  # ONNX export
onnxruntime>=1.9.0
onnx-simplifier>=0.3.6  # ONNX simplifier
# tensorflow>=2.4.1  # TFLite export
# tensorflowjs>=3.9.0  # TF.js export
# openvino-dev  # OpenVINO export

# Extras --------------------------------------
albumentations>=1.0.3
# roboflow
thop  # FLOPs computation
pymediainfo