annotations - 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

尝试这个：

python detect.py img.jpg --save_txt

如果你想按类过滤检测，试试这个

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

该示例将保留属于类 0、2 和 3 的那些对象。