ai-lab
ai-lab copied to clipboard
[FEATURE] Detatch confidence from labels
Currently if no labels are passed, the confidence is not show, only the bounding box.
So this code:
<AILabLocalVideo
ObjectDetectionUI={AILabObjectDetectionUI}
model={model}
modelConfig={{
modelType: 'ssd',
// labels: CLASSES,
}}
visual
src={'/walkers.mov'}
/>
creates this result:

But when you add labels, the label AND the confidence are displayed.
So this code:
<AILabLocalVideo
ObjectDetectionUI={AILabObjectDetectionUI}
model={model}
modelConfig={{
modelType: 'ssd',
labels: CLASSES,
}}
visual
src={'/walkers.mov'}
/>
creates this result:

It would be cool if modelConfig had a setting for "confidence: true" that would show or hide the confidence regardless of the labels.