是否有使用自定义权重的 labeling/annotating 图像工具？

Question

我在 YOLOv5 上训练了一个模型来检测自定义对象。是否有工具可以使用这些权重来标记显示相同 类 的新图像？ 谢谢！

Answer 1

试试这个：

python detect.py img.jpg --save_txt

如果您想按 class 过滤检测，试试这个

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

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