Add requirements.txt file and update README file

isl-org · Jul 7, 2021 · 12e684e · 12e684e
1 parent 3a24773
commit 12e684e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -25,7 +25,7 @@ Segmentation:
 2) Set up dependencies: 
 
     ```shell
-    pip install torch torchvision opencv-python timm==0.4.5
+    pip install -r requirements.txt
     ```
 
    The code was tested with Python 3.7, PyTorch 1.8.0, OpenCV 4.5.1, and timm 0.4.5

diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,4 @@
+torch==1.8.0
+torchvision==0.9.0
+opencv-python==4.5.2.54
+timm==0.4.5