{"id":1794,"date":"2021-01-15T19:47:54","date_gmt":"2021-01-15T11:47:54","guid":{"rendered":"http:\/\/www.yatenglg.cn\/?p=1794"},"modified":"2022-10-18T16:44:35","modified_gmt":"2022-10-18T08:44:35","slug":"%e7%94%bb%e6%a1%86-cv2","status":"publish","type":"post","link":"http:\/\/www.yatenglg.cn\/blog\/?p=1794","title":{"rendered":"\u753b\u6846-cv2"},"content":{"rendered":"\n<div class=\"hcb_wrap\"><pre class=\"prism undefined-numbers lang-python\" data-file=\"draw_boxes_cv2.py\" data-lang=\"Python\"><code>def draw_boxs(img:Union[str, np.ndarray],\n              boxes:np.ndarray,\n              labels:np.ndarray = None,\n              label_name:Dict[int, str]=None,\n              font_scale:int = 1,\n              line_thickness:int = 1):\n    &quot;&quot;&quot;\n\n    :param img:             img path or img date, shape: [h, w ,c]\n    :param boxes:           [xmin, ymin, xmax, ymax], shape: [n, 4],\n    :param labels:          shape: [n]\n    :param label_name:      dict, {index, name}\n    :param font_scale:      int\n    :param line_thickness:  int\n    :return:\n    &quot;&quot;&quot;\n    if labels is not None:\n        assert boxes.shape[0] == labels.shape[0]\n\n    if isinstance(img, str):\n        img = cv2.imread(img)\n\n    for i in range(boxes.shape[0]):\n\n        color = (0, 255, 0)\n        if labels is not None:\n            label = labels[i]\n            color = [int(i * (label ** 2 - label)) % 255 for i in (15, 127, 7)]\n            print(color)\n            if label_name is not None:\n                if label in label_name:\n                    label = label_name[label]\n            (text_w, text_h), _ = cv2.getTextSize(&#39;{}&#39;.format(label), cv2.FONT_HERSHEY_COMPLEX_SMALL,\n                                                  font_scale, font_scale)\n            zero = np.zeros((img.shape), dtype=np.uint8)\n            background = cv2.rectangle(zero, (boxes[i][0], boxes[i][3]-text_h),\n                                       (boxes[i][0]+text_w, boxes[i][3]), color, -1)\n            img = cv2.addWeighted(img, 1, background, 1, 1)\n            cv2.putText(img, &#39;{}&#39;.format(label), (boxes[i][0], boxes[i][3]),\n                        cv2.FONT_HERSHEY_COMPLEX_SMALL, font_scale, (0, 0, 0), font_scale)\n        cv2.rectangle(img, (boxes[i][0], boxes[i][1]), (boxes[i][2], boxes[i][3]), color, line_thickness)\n    return img<\/code><\/pre><\/div>\n","protected":false},"excerpt":{"rendered":"","protected":false},"author":1,"featured_media":0,"comment_status":"closed","ping_status":"closed","sticky":false,"template":"","format":"standard","meta":[],"categories":[7],"tags":[],"_links":{"self":[{"href":"http:\/\/www.yatenglg.cn\/blog\/index.php?rest_route=\/wp\/v2\/posts\/1794"}],"collection":[{"href":"http:\/\/www.yatenglg.cn\/blog\/index.php?rest_route=\/wp\/v2\/posts"}],"about":[{"href":"http:\/\/www.yatenglg.cn\/blog\/index.php?rest_route=\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"http:\/\/www.yatenglg.cn\/blog\/index.php?rest_route=\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"http:\/\/www.yatenglg.cn\/blog\/index.php?rest_route=%2Fwp%2Fv2%2Fcomments&post=1794"}],"version-history":[{"count":5,"href":"http:\/\/www.yatenglg.cn\/blog\/index.php?rest_route=\/wp\/v2\/posts\/1794\/revisions"}],"predecessor-version":[{"id":1799,"href":"http:\/\/www.yatenglg.cn\/blog\/index.php?rest_route=\/wp\/v2\/posts\/1794\/revisions\/1799"}],"wp:attachment":[{"href":"http:\/\/www.yatenglg.cn\/blog\/index.php?rest_route=%2Fwp%2Fv2%2Fmedia&parent=1794"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"http:\/\/www.yatenglg.cn\/blog\/index.php?rest_route=%2Fwp%2Fv2%2Fcategories&post=1794"},{"taxonomy":"post_tag","embeddable":true,"href":"http:\/\/www.yatenglg.cn\/blog\/index.php?rest_route=%2Fwp%2Fv2%2Ftags&post=1794"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}