Update requirements

requirements.txt

@@ -1,11 +1,15 @@
-pytest==3.9.1
-flake8==3.5.0
-numpy==1.15.2
-tensorflow-gpu==1.11.0
-scipy==1.1.0
-Pillow==5.3.0
-requests==2.20.0
-googledrivedownloader==0.3
-tqdm==4.28.0
-pyside2==5.15.2
-opencv-python==4.5.1.48
+matplotlib>=3.2.2
+numpy>=1.18.5
+opencv-python>=4.1.2
+Pillow
+PyYAML>=5.3.1
+scipy>=1.4.1
+torch==1.8.1
+torchaudio===0.8.1
+torchvision>=0.8.1
+tqdm>=4.41.0
+pyside2>=5.15.2
+requests>=2.25.1
+pandas>=1.2.3
+seaborn>=0.11.1
+tensorboard>=2.4.1

requirements_gpu.txt

@@ -0,0 +1,17 @@
+matplotlib>=3.2.2
+numpy>=1.18.5
+opencv-python>=4.1.2
+Pillow
+PyYAML>=5.3.1
+scipy>=1.4.1
+torch==1.8.1+cu111
+torchaudio===0.8.1
+torchvision>=0.8.1
+tqdm>=4.41.0
+pyside2>=5.15.2
+requests>=2.25.1
+pandas>=1.2.3
+seaborn>=0.11.1
+tensorboard>=2.4.1
+
+--find-links https://download.pytorch.org/whl/torch_stable.html