xialuxi / yolov5-car-plate

基于yolov5的车牌检测,包含车牌角点检测
GNU General Public License v3.0
152 stars 40 forks source link

关于SIOU代码 #17

Closed ChristmasLee closed 2 years ago

ChristmasLee commented 2 years ago

作者您好!在您SIOU代码里面有一部分是:

if GIoU or DIoU or CIoU:
cw = torch.max(b1_x2, b2_x2) - torch.min(b1_x1, b2_x1) # convex (smallest enclosing box) width
ch = torch.max(b1_y2, b2_y2) - torch.min(b1_y1, b2_y1) # convex height
     if SIoU: # SIoU Loss https://arxiv.org/pdf/2205.12740.pdf
s_cw = (b2_x1 + b2_x2 - b1_x1 - b1_x2) * 0.5

第一个if是不是要加上一个or SIOU啊