From bcb2b3d6c8505957dfe0705a5173fbb2e84142c9 Mon Sep 17 00:00:00 2001
From: Sven Peter <sven.peter@iwr.uni-heidelberg.de>
Date: Sun, 1 Sep 2019 12:01:50 +0200
Subject: [PATCH] fix setup.py requirements

---
 setup.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index e2a6b5e1..a2d00ba2 100755
--- a/setup.py
+++ b/setup.py
@@ -23,7 +23,9 @@
     "scipy>=0.13.0",
     "h5py",
     "numpy>=1.8",
-    "scikit-image"
+    "scikit-image",
+    "torchvision",
+    "tqdm"
 ]