search

HumanSignal / label-studio-converter

Tools for converting Label Studio annotations into common dataset formats
https://labelstud.io/
255 stars 132 forks source link

fix: import Yolo predictions (detection task with bounding boxes, labels with prediction confidence) #256

Closed carbaro closed 4 months ago

carbaro commented 10 months ago

This pull requests simply accounts for prediction labels produced by yolo's --save_conf option

niklub commented 4 months ago

The change has been already introduced in https://github.com/HumanSignal/label-studio-converter/blob/master/label_studio_converter/imports/yolo.py#L116

carbaro commented 3 months ago

@niklub Thanks for reviewing this PR and letting me know of the alternative PR accepted for this purpose.

I did find that it imports predictions as confirmed detections (as if a user had created and submitted the annotations).

I've created #295 which warns users to use the --out_type if prediction scores are detected while importing with the default value "annotations".