Merge pull request #5 from KiLJ4EdeN/master

omasaht · web-flow · commit 002549c981b6 · 2020-11-21T15:14:10.000+02:00
Add requirements.txt file for easier dependency handling
diff --git a/README.md b/README.md
@@ -45,6 +45,14 @@ torchvision               0.6.1
 onnx                      1.7.0
 onnxruntime               1.2.0
 ```
+
+
+Installation with pip
+```bash
+pip3 install -r requirements.txt
+```
+
+
 You may also need to install jupyter to access notebooks (.ipynb). It is recommended that you use Anaconda to install packages.
 
 Code has been tested on Ubuntu 18.04
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,9 @@
+numpy==1.18.5
+opencv-python==4.2.0.32
+opencv-contrib-python==4.2.0.32
+scipy==1.5.0
+matplotlib==3.2.2
+torch==1.5.1
+torchvision==0.6.1
+onnx==1.7.0
+onnxruntime==1.2.0
