yolov2_demo.py 文件源码

def draw_result(img, result):
    for i in range(len(result)):
        left = int(result[i][1])
        right = int(result[i][2])
        top = int(result[i][3])
        bot = int(result[i][4])
        c = i%3
        color = 200*(c==0), 200*(c==1), 200*(c==2)
        cv2.rectangle(img, (left, top), (right, bot), (color), 5)
        cv2.rectangle(img, (left, top + 20),
                      (right, top+1), (color), -1)
        cv2.putText(img, result[i][0] + ' : %.2f' % result[i][5], (left+ 15, top -7 + 20), cv2.FONT_HERSHEY_COMPLEX, 0.5, (255, 255, 255), 1, cv2.CV_AA)