You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: src/deepsparse/yolo/README.md
+2-2
Original file line number
Diff line number
Diff line change
@@ -37,7 +37,7 @@ sparseml.yolov5.export_onnx \
37
37
--weights path/to/your/model \
38
38
--dynamic #Allows for dynamic input shape
39
39
```
40
-
This creates `model.onnx` file, in the directory of your `weights`(e.g. `runs/train/weights/model.onnx`).
40
+
This creates a DeepSparse_Deployment folder with a `model.onnx` file (e.g. `runs/train/exp/DeepSparse_Deployment/model.onnx`).
41
41
42
42
#### SparseZoo Stub
43
43
Alternatively, you can skip the process of the ONNX model export by using Neural Magic's [SparseZoo](https://sparsezoo.neuralmagic.com/). The SparseZoo contains pre-sparsified models and SparseZoo stubs enable you to reference any model on the SparseZoo in a convenient and predictable way.
The following example uses pipelines to run a pruned and quantized YOLOv5l model for inference, downloaded by default from the SparseZoo. As input the pipeline ingests a list of images and returns for each image the detection boxes in numeric form.
0 commit comments