infer num_classes for yolov8-seg pipeline

neuralmagic / deepsparse

Sparsity-aware deep learning inference runtime for CPUs

https://neuralmagic.com/deepsparse/

Other

3.01k stars 176 forks source link

Closed bfineran closed 1 year ago

bfineran commented 1 year ago

non default num_classes not currently supported. this PR adds a quick patch to infer based on the scores dimension

test_plan: manual by @rsnm2

bfineran commented 1 year ago

failures unrelated