Search code examples
annotationsyolodataset

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!

Solution

  • 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.