requirements.txt

# dicom
python-gdcm==3.0.9.1
pydicom==2.1.2

# joblib
joblib==1.0.1

# tensorflow
tensorflow==2.4.1

# torch
torch==1.7.0
torchvision==0.8.1

# base
numpy==1.19.5
pandas==1.2.4
matplotlib==3.4.2
opencv-python==4.5.2.54
opencv-python-headless==4.5.2.54
Pillow==8.2.0
PyYAML>=5.3.1
scipy==1.6.3
tqdm==4.61.1

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

# plotting ------------------------------------
seaborn==0.11.1

# weighted boxes fusion
ensemble_boxes==1.0.6

# Albumentation
albumentations==1.0.1

# export --------------------------------------
# coremltools>=4.1
# onnx>=1.9.0
# scikit-learn==0.19.2  # for coreml quantization

# extras --------------------------------------
thop==0.0.31.post2005241907  # FLOPS computation
Cython==0.29.23  # for pycocotools https://github.com/cocodataset/cocoapi/issues/172
pycocotools==2.0  # COCO mAP

# flexible
addict==2.4.0
timm==0.4.12

# efficientnet
efficientnet==1.1.1