记录一些python的使用

1.从list中取出每个int元素,用0补为固定长度,形成一个字符串:

box_str="-".join(['0'*(4-len(str(x)))+str(x) for x in position])

2.用来计算两个矩形overlap的函数:

def mat_inter(box1, box2):
# 判断两个矩形是否相交
# box=(xA,yA,xB,yB)
x01, y01, x02, y02 = box1
x11, y11, x12, y12 = box2

lx = abs((x01 + x02) / 2 - (x11 + x12) / 2)
ly = abs((y01 + y02) / 2 - (y11 + y12) / 2)
sax = abs(x01 - x02)
sbx = abs(x11 - x12)
say = abs(y01 - y02)
sby = abs(y11 - y12)
if lx <= (sax + sbx) / 2 and ly <= (say + sby) / 2:
    return True
else:
    return False


def solve_coincide(box1, box2):
    # box=[xA,yA,xB,yB]
    # 计算两个矩形框的重合度
    x01, y01, x02, y02 = box1
    x11, y11, x12, y12 = box2
    col = min(x02, x12) - max(x01, x11)
    row = min(y02, y12) - max(y01, y11)
    intersection = col * row

    area1 = (x02 - x01) * (y02 - y01)
    area2 = (x12 - x11) * (y12 - y11)
    # coincide = float(intersection / (area1 + area2 - intersection))
    coincide=format(float(intersection) / float((area1 + area2 - intersection)), '.5f') #保留5位小数
    return coincide

你可能感兴趣的:(代码记录)