update YOLO examples dependencies (#118)

* update YOLO examples dependencies * create requirements.txt for YOLO examples
neuralmagic · May 28, 2021 · 16c7915 · 16c7915
1 parent 5a5557f
commit 16c7915
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 1 deletion.
diff --git a/examples/README.md b/examples/README.md
@@ -30,3 +30,4 @@ Open a Pull Request to [contribute](https://github.com/neuralmagic/deepsparse/bl
 | [Classification](https://github.com/neuralmagic/deepsparse/blob/main/examples/classification/)  | How to use classification models from SparseZoo to perform inference and benchmarking with the DeepSparse Engine  |
 | [Detection](https://github.com/neuralmagic/deepsparse/blob/main/examples/detection/)  | How to use object detection models from SparseZoo to perform inference and benchmarking with the DeepSparse Engine  |
 | [Model Server](https://github.com/neuralmagic/deepsparse/blob/main/examples/flask/)  | Simple model server and client example, showing how to use the DeepSparse Engine as an inference backend for a real-time inference server |
+| [YOLOv3](https://github.com/neuralmagic/deepsparse/blob/main/examples/ultralytics-yolov3/) | Serving, benchmarking, and running annotation inferences with YOLOv3 models |
diff --git a/examples/ultralytics-yolov3/README.md b/examples/ultralytics-yolov3/README.md
@@ -27,7 +27,7 @@ create your own using the
 ## Installation
 The dependencies for this example can be installed using `pip`:
 ```bash
-pip3 install deepsparse sparseml[torchvision] flask flask-cors
+pip3 install -r requirements.txt
 ```
 
 ## Annotation Example

diff --git a/examples/ultralytics-yolov3/requirements.txt b/examples/ultralytics-yolov3/requirements.txt
@@ -0,0 +1,10 @@
+# pip install -r requirements.txt
+
+# NM Deps
+deepsparse>=0.3.1
+sparseml[torchvision]>=0.3.1
+
+# Example Deps
+flask
+flask-cors
+opencv-python