diff --git a/README.md b/README.md index 2f90b4f..1e95735 100644 --- 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 new file mode 100644 index 0000000..00d5e94 --- /dev/null +++ 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