Merge pull request #5 from KiLJ4EdeN/master

Add requirements.txt file for easier dependency handling
omasaht · Nov 21, 2020 · 002549c · 002549c
2 parents c6af2cc + d9e76b8
commit 002549c
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
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