SwimmingLiu/YOLOSHOW

log_string += f"{n} {self.names[int(c)]}{'s' * (n > 1)}, " KeyError: 15

llllaaa519 opened this issue · 5 comments

llllaaa519 commented

Traceback (most recent call last):
File "D:\project\YOLOSHOW-master\yolocode\yolov8\YOLOv8Thread.py", line 105, in run
self.detect()
File "D:\project\YOLOSHOW-master\yolocode\yolov8\YOLOv8Thread.py", line 190, in detect
label_str = self.write_results(i, self.results, (p, im, im0)) # labels /// original :s +=
File "D:\project\YOLOSHOW-master\yolocode\yolov8\YOLOv8Thread.py", line 374, in write_results
log_string += result.verbose()
File "D:\project\YOLOSHOW-master\yolocode\yolov8\engine\results.py", line 327, in verbose
log_string += f"{n} {self.names[int(c)]}{'s' * (n > 1)}, "
KeyError: 15
我使用了官方的OBB检测权重进行旋转目标检测，报了这个错误。

SwimmingLiu commented

目前暂不支持obb检测模式，后续会考虑补上该功能

llllaaa519 commented

好的感谢回答，请问一下想自己添加这个功能大概怎么添加呢

SwimmingLiu commented
  1. 拉取最新的ultralytics仓库YOLOv8代码，然后找到models/yolo/obb/predict.py
  2. 参考YOLOv8Thread.py 改成 YOLOv8ObbThread.py
  3. YOLOSHOW对应添加YOLOv8ObbThread的内容
llllaaa519 commented
  1. 拉取最新的ultralytics仓库YOLOv8代码，然后找到models/yolo/obb/predict.py
  2. 参考YOLOv8Thread.py 改成 YOLOv8ObbThread.py
  3. YOLOSHOW对应添加YOLOv8ObbThread的内容
    好的，谢谢，我试试
llllaaa519 commented

我按照你的步骤改了一下，现在已经可以实现OBB检测了，但没有统计类别数和目标数，请问一下是不是遗漏了某部分代码没有改，请问一下统计类别数和目标数的代码在哪部分啊，谢谢