From 92799a67daeac6a2bbff8674d6d42b5d3c8c2964 Mon Sep 17 00:00:00 2001 From: Tobias Eidelpes Date: Thu, 16 Feb 2023 14:10:04 +0100 Subject: [PATCH] Add additional requirements --- code/requirements.txt | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/code/requirements.txt b/code/requirements.txt index 17fbed6..77d9bc2 100644 --- a/code/requirements.txt +++ b/code/requirements.txt @@ -3,9 +3,22 @@ apscheduler==3.10.0 albumentations==1.3.0 pandas==1.1.5 onnxruntime==1.8.0 +opencv-python==4.7.0 +torch==1.8.1 +torchvision==0.9.1 +numpy==1.18.0 +scipy==1.1.0 +scikit-learn==0.21.3 +PIL==8.0.4 +argparse==1.1 +json==2.0.9 +matplotlib==3.3.4 # For labeling/conversion # label-studio==1.7.0 # For interactive evaluation # fiftyone + +# Visualize gradients +# pytorch-gradcam