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

Question

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!

Answer 1

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.