CODEWITHSUNDEEP
annotationsyolodataset
st0412
st0412

Reputation: 49

Is there a tool for labeling/annotating images using custom weights?

I´ve trained a model on YOLOv5 for detecting custom objects. Is there a tool to use those weights to label new images showing the same classes? Thanks!

Upvotes: 2

Views: 457

Answers (1)

Illustrati
Illustrati

Reputation: 321

Try this:

python detect.py img.jpg --save_txt

If you want to filter detections by class, try this

python detect.py img.jpg --save_txt --class 0 2 3

The example will keep those objects belonging to classes 0, 2, and 3.

Upvotes: 1

Related Questions