Egrt / yolov7-obb

在YOLOv7的基础上使用KLD损失修改为旋转目标检测yolov7-obb
GNU General Public License v3.0
176 stars 26 forks source link

偵測框的粗細 #26

Closed chishiwen closed 1 year ago

chishiwen commented 1 year ago

@Egrt 作者您好 想請教您 若要修改BBOX thickness,是直接修改yolo.py 165 行中的何值。

yolo.py --> 165 thickness = int(max((image.size[0] + image.size[1]) // np.mean(self.input_shape), 1))

試過 改成 thickness = 2 或10,儲存後進行偵測(predict.py),但BBOX大小無改變 也試過 改成 thickness = int(max((image.size[0] + image.size[1]) // np.mean(self.input_shape), 2)),儲存後進行偵測(predict.py),BBOX大小仍無改變

想請問我有哪個環節少做,或是做錯嗎,謝謝。

chishiwen commented 1 year ago

後來自己找到 在yolo.py 195行 增加", width=X" ex : 195 | draw.polygon(xy=polygon_list, outline=self.colors[c],width=2)