Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Summary
process_engine_outputs
has a slight change. Previously providingnm
for number of masks, which is no longer used by the yolov8non_max_suppression
function.Testing:
sparsezoo models tested:
"zoo:cv/detection/yolov8-m/pytorch/ultralytics/coco/base_quant-none"
"zoo:cv/segmentation/yolov8-n/pytorch/ultralytics/coco/base-none"
yolov8 models tested:
yolov8n.pt
yolov8n-seg.pt
Pipeline creation + forward pass:
Detection
Segmentation:
Annotation:
Detection:
deepsparse.yolov8.annotate --source basilica.jpg --model_filepath "yolov8n.onnx"
Segmentation:
deepsparse.yolov8.annotate --source basilica.jpg --model_filepath "yolov8n-seg.onnx" --subtask segmentation
Results verified for outputs produced by both workflows above
Eval
Detection:
deepsparse.yolov8.eval --model_path yolov8n.onnx
Segementation:
deepsparse.yolov8.eval --model_path yolov8n-seg.onnx --subtask segmentation