{"id":5147,"date":"2022-08-02T17:19:14","date_gmt":"2022-08-02T09:19:14","guid":{"rendered":"http:\/\/139.9.1.231\/?p=5147"},"modified":"2022-08-02T17:38:05","modified_gmt":"2022-08-02T09:38:05","slug":"yolov7-pose","status":"publish","type":"post","link":"http:\/\/139.9.1.231\/index.php\/2022\/08\/02\/yolov7-pose\/","title":{"rendered":"YOLOv7-Pose  \u57fa\u4e8eYOLOv7\u7684\u5173\u952e\u70b9\u6a21\u578b"},"content":{"rendered":"\n<p class=\"has-normal-font-size\">        \u76ee\u524d\u4eba\u4f53\u59ff\u6001\u4f30\u8ba1\u603b\u4f53\u5206\u4e3aTop-down\u548cBottom-up\u4e24\u79cd\uff0c\u4e0e\u76ee\u6807\u68c0\u6d4b\u4e0d\u540c\uff0c\u65e0\u8bba\u662f\u57fa\u4e8e\u70ed\u529b\u56fe\u6216\u662f\u57fa\u4e8e\u68c0\u6d4b\u5668\u5904\u7406\u7684\u5173\u952e\u70b9\u68c0\u6d4b\u7b97\u6cd5\uff0c\u90fd\u8f83\u4e3a\u4f9d\u8d56\u8ba1\u7b97\u8d44\u6e90\uff0c\u63a8\u7406\u8017\u65f6\u7565\u957f\uff0c\u4eca\u5e74\u51fa\u73b0\u4e86\u4ee5YOLO\u4e3a\u57fa\u7ebf\u7684\u5173\u952e\u70b9\u68c0\u6d4b\u5668\u3002\u73a9\u8fc7\u76ee\u6807\u68c0\u6d4b\u7684\u7ae5\u978b\u90fd\u77e5\u9053YOLO\u4ee5\u53ca\u5404\u79cd\u53d8\u79cd\u76ee\u524d\u7b97\u662f\u5de5\u4e1a\u843d\u5730\u8f83\u591a\u7684\u4e00\u7c7b\u68c0\u6d4b\u5668\uff0c\u5176\u7b80\u5355\u7684\u8bbe\u8ba1\u601d\u60f3\uff0c\u957f\u671f\u6d3b\u8dc3\u7684\u793e\u533a\u751f\u6001\uff0c\u4f7f\u5176\u59cb\u7ec8\u5360\u636e\u7740\u8f83\u9ad8\u7684\u8bdd\u9898\u5ea6\u3002<\/p>\n\n\n\n<h3>\u3010\u6f14\u53d8\u3011<\/h3>\n\n\n\n<p class=\"has-normal-font-size\">\u5728ECCV 2022\u548cCVPRW 2022\u4f1a\u8bae\u4e0a\uff0cYoLo-Pose\u548cKaPao\uff08\u4e0b\u79f0\u4e3ayolo-like-pose\uff09\u90fd\u57fa\u4e8e\u6d41\u884c\u7684YOLO\u76ee\u6807\u68c0\u6d4b\u6846\u67b6\u63d0\u51fa\u4e00\u79cd\u65b0\u9896\u7684\u65e0\u70ed\u529b\u56fe\u7684\u65b9\u6cd5\uff0c\u7c7b\u4f3c\u4e8e\u5f88\u4e45\u4ee5\u524d\u8c37\u6b4c\u4f7f\u7528\u56de\u5f52\u8ba1\u7b97\u5173\u952e\u70b9\u7684\u601d\u60f3\uff0cyolo-like-pose\u4e00\u4e0d\u4f7f\u7528\u68c0\u6d4b\u5668\u8fdb\u884c\u4e8c\u9636\u5904\u7406\uff0c\u4e8c\u90e8\u4f7f\u7528\u70ed\u529b\u56fe\u62fc\u63a5\uff0c\u867d\u7136\u662f\u4e00\u79cd\u66b4\u529b\u56de\u5f52\u5173\u952e\u70b9\u7684\u68c0\u6d4b\u7b97\u6cd5\uff0c\u4f46\u5728\u5904\u7406\u901f\u5ea6\u4e0a\u5177\u6709\u4e00\u5b9a\u4f18\u52bf\u3002<\/p>\n\n\n\n<figure class=\"wp-block-image size-full\"><img loading=\"lazy\" width=\"319\" height=\"164\" src=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-9.png\" alt=\"\" class=\"wp-image-5154\" srcset=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-9.png 319w, http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-9-300x154.png 300w\" sizes=\"(max-width: 319px) 100vw, 319px\" \/><\/figure>\n\n\n\n<h4>kapao<\/h4>\n\n\n\n<p class=\"has-normal-font-size\">\u53bb\u5e7411\u6708\uff0c\u6ed1\u94c1\u5362\u5927\u5b66\u7387\u5148\u63d0\u51fa\u4e86 KaPao\uff1aRethinking Keypoint Representations: Modeling Keypoints and Poses as Objects for Multi-Person Human Pose Estimation\uff0c\u57fa\u4e8eYOLOv5\u8fdb\u884c\u5173\u952e\u70b9\u68c0\u6d4b\uff0c\u8be5\u6587\u7ae0\u76ee\u524d\u5df2\u88abECCV 2022\u63a5\u6536\uff0c\u8be5\u7b97\u6cd5\u6240\u53d6\u5f97\u7684\u6027\u80fd\u5982\u4e0b\uff1a<\/p>\n\n\n\n<figure class=\"wp-block-image size-full\"><img loading=\"lazy\" width=\"680\" height=\"515\" src=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-10.png\" alt=\"\" class=\"wp-image-5155\" srcset=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-10.png 680w, http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-10-300x227.png 300w\" sizes=\"(max-width: 680px) 100vw, 680px\" \/><\/figure>\n\n\n\n<p class=\"has-normal-font-size\">paper\uff1ahttps:\/\/arxiv.org\/abs\/2111.08557<\/p>\n\n\n\n<p class=\"has-normal-font-size\">code\uff1ahttps:\/\/github.com\/wmcnally\/kapao<\/p>\n\n\n\n<h4>yolov5-pose<\/h4>\n\n\n\n<p class=\"has-normal-font-size\">\u4eca\u5e744\u6708\uff0cyolo-pose\u4e5f\u6302\u5728\u4e86arvix\uff0c\u5728\u8bba\u6587\u4e2d\uff0c\u901a\u8fc7\u8c03\u7814\u53d1\u73b0 HeatMap \u7684\u65b9\u5f0f\u666e\u904d\u4f7f\u7528L1 Loss\u3002\u7136\u800c\uff0cL1\u635f\u5931\u5e76\u4e0d\u4e00\u5b9a\u9002\u5408\u83b7\u5f97\u6700\u4f73\u7684OKS\u3002\u4e14\u7531\u4e8eHeatMap\u662f\u6982\u7387\u56fe\uff0c\u56e0\u6b64\u5728\u57fa\u4e8e\u7eafHeatMap\u7684\u65b9\u6cd5\u4e2d\u4e0d\u53ef\u80fd\u4f7f\u7528OKS\u4f5c\u4e3aloss\uff0c\u53ea\u6709\u5f53\u56de\u5f52\u5230\u5173\u952e\u70b9\u4f4d\u7f6e\u65f6\uff0cOKS\u624d\u80fd\u88ab\u7528\u4f5c\u635f\u5931\u51fd\u6570\u3002\u56e0\u6b64\uff0cyolo-pose\u4f7f\u7528oks loss\u4f5c\u4e3a\u5173\u952e\u70b9\u7684\u635f\u5931<\/p>\n\n\n\n<figure class=\"wp-block-image size-full\"><img loading=\"lazy\" width=\"1003\" height=\"616\" src=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-11.png\" alt=\"\" class=\"wp-image-5157\" srcset=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-11.png 1003w, http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-11-300x184.png 300w, http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-11-768x472.png 768w\" sizes=\"(max-width: 1003px) 100vw, 1003px\" \/><\/figure>\n\n\n\n<p class=\"has-normal-font-size\">\u76f8\u5173\u4ee3\u7801\u5728https:\/\/github.com\/TexasInstruments\/edgeai-yolov5\/blob\/yolo-pose\/utils\/loss.py\u4e5f\u53ef\u89c1\u5230\uff1a<\/p>\n\n\n\n<p>&nbsp;if&nbsp;self.kpt_label:<br>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;#Direct&nbsp;kpt&nbsp;prediction<br>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;pkpt_x&nbsp;=&nbsp;ps[:,&nbsp;6::3]&nbsp;*&nbsp;2.&nbsp;&#8211;&nbsp;0.5<br>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;pkpt_y&nbsp;=&nbsp;ps[:,&nbsp;7::3]&nbsp;*&nbsp;2.&nbsp;&#8211;&nbsp;0.5<br>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;pkpt_score&nbsp;=&nbsp;ps[:,&nbsp;8::3]<br>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;#mask<br>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;kpt_mask&nbsp;=&nbsp;(tkpt[i][:,&nbsp;0::2]&nbsp;!=&nbsp;0)<br>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;lkptv&nbsp;+=&nbsp;self.BCEcls(pkpt_score,&nbsp;kpt_mask.float())&nbsp;<br>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;#l2&nbsp;distance&nbsp;based&nbsp;loss<br>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;#lkpt&nbsp;+=&nbsp;(((pkpt-tkpt[i])*kpt_mask)**2).mean()&nbsp;&nbsp;#Try&nbsp;to&nbsp;make&nbsp;this&nbsp;loss&nbsp;based&nbsp;on&nbsp;distance&nbsp;instead&nbsp;of&nbsp;ordinary&nbsp;difference<br>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;#oks&nbsp;based&nbsp;loss<br>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;d&nbsp;=&nbsp;(pkpt_x-tkpt[i][:,0::2])**2&nbsp;+&nbsp;(pkpt_y-tkpt[i][:,1::2])**2<br>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;s&nbsp;=&nbsp;torch.prod(tbox[i][:,-2:],&nbsp;dim=1,&nbsp;keepdim=True)<br>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;kpt_loss_factor&nbsp;=&nbsp;(torch.sum(kpt_mask&nbsp;!=&nbsp;0)&nbsp;+&nbsp;torch.sum(kpt_mask&nbsp;==&nbsp;0))\/torch.sum(kpt_mask&nbsp;!=&nbsp;0)<br>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;lkpt&nbsp;+=&nbsp;kpt_loss_factor*((1&nbsp;&#8211;&nbsp;torch.exp(-d\/(s*(4*sigmas**2)+1e-9)))*kpt_mask).mean()<\/p>\n\n\n\n<figure class=\"wp-block-image size-full\"><img loading=\"lazy\" width=\"1016\" height=\"495\" src=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-12.png\" alt=\"\" class=\"wp-image-5158\" srcset=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-12.png 1016w, http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-12-300x146.png 300w, http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-12-768x374.png 768w\" sizes=\"(max-width: 1016px) 100vw, 1016px\" \/><\/figure>\n\n\n\n<h4>yolov7-pose<\/h4>\n\n\n\n<p class=\"has-normal-font-size\">\u4e0a\u4e2a\u661f\u671f\uff0cYOLOv7\u7684\u4f5c\u8005\u4e5f\u653e\u51fa\u4e86\u5173\u4e8e\u4eba\u4f53\u5173\u952e\u70b9\u68c0\u6d4b\u7684\u6a21\u578b\uff0c\u8be5\u6a21\u578b\u57fa\u4e8eYOLOv7-w6<\/p>\n\n\n\n<figure class=\"wp-block-image size-full\"><img loading=\"lazy\" width=\"970\" height=\"364\" src=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-13.png\" alt=\"\" class=\"wp-image-5159\" srcset=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-13.png 970w, http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-13-300x113.png 300w, http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-13-768x288.png 768w\" sizes=\"(max-width: 970px) 100vw, 970px\" \/><\/figure>\n\n\n\n<p class=\"has-normal-font-size\">\u76ee\u524d\u4f5c\u8005\u63d0\u4f9b\u4e86.pt\u6587\u4ef6\u548c\u63a8\u7406\u6d4b\u8bd5\u7684\u811a\u672c\uff0c\u6709\u5174\u8da3\u7684\u7ae5\u9774\u53ef\u4ee5\u53bb\u770b\u770b\uff0c\u672c\u6587\u7684\u91cd\u70b9\u66f4\u504f\u5411\u4e8e\u5bf9yolov7-pose.pt\u8fdb\u884connx\u6587\u4ef6\u7684\u62bd\u53d6\u548c\u63a8\u7406\u3002<\/p>\n\n\n\n<h3>\u3010yolov7-pose + onnxruntime\u3011<\/h3>\n\n\n\n<p class=\"has-normal-font-size\">\u9996\u5148\u4e0b\u8f7d\u597d\u5b98\u65b9\u7684\u9884\u8bad\u7ec3\u6a21\u578b\uff0c\u4f7f\u7528\u63d0\u4f9b\u7684\u811a\u672c\u8fdb\u884c\u63a8\u7406\uff1a<\/p>\n\n\n\n<pre class=\"wp-block-code\"><code>%&nbsp;weigths&nbsp;=&nbsp;torch.load('weights\/yolov7-w6-pose.pt')<br>%&nbsp;image&nbsp;=&nbsp;cv2.imread('sample\/pose.jpeg')<br>!python&nbsp;pose.py&nbsp;<\/code><\/pre>\n\n\n\n<p class=\"has-normal-font-size\"><strong>\u4e00\u3001yolov7-w6 VS yolov7-w6-pose<\/strong>\uff1a<\/p>\n\n\n\n<p class=\"has-normal-font-size\">\u9996\u5148\u770b\u4e0byolov7-w6\u4f7f\u7528\u7684\u68c0\u6d4b\u5934<\/p>\n\n\n\n<figure class=\"wp-block-image size-full\"><img loading=\"lazy\" width=\"715\" height=\"717\" src=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-14.png\" alt=\"\" class=\"wp-image-5161\" srcset=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-14.png 715w, http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-14-300x300.png 300w, http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-14-150x150.png 150w, http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-14-120x120.png 120w\" sizes=\"(max-width: 715px) 100vw, 715px\" \/><\/figure>\n\n\n\n<figure class=\"wp-block-image size-full\"><img loading=\"lazy\" width=\"748\" height=\"174\" src=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-15.png\" alt=\"\" class=\"wp-image-5162\" srcset=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-15.png 748w, http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-15-300x70.png 300w\" sizes=\"(max-width: 748px) 100vw, 748px\" \/><\/figure>\n\n\n\n<figure class=\"wp-block-image size-full\"><img loading=\"lazy\" width=\"726\" height=\"932\" src=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-16.png\" alt=\"\" class=\"wp-image-5163\" srcset=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-16.png 726w, http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-16-234x300.png 234w\" sizes=\"(max-width: 726px) 100vw, 726px\" \/><\/figure>\n\n\n\n<figure class=\"wp-block-image size-full\"><img loading=\"lazy\" width=\"540\" height=\"97\" src=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-17.png\" alt=\"\" class=\"wp-image-5164\" srcset=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-17.png 540w, http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-17-300x54.png 300w\" sizes=\"(max-width: 540px) 100vw, 540px\" \/><\/figure>\n\n\n\n<p><strong>\u4e8c\u3001\u4fee\u6539export\u811a\u672c<\/strong><\/p>\n\n\n\n<p class=\"has-normal-font-size\">\u5982\u679c\u76f4\u63a5\u4f7f\u7528export\u811a\u672c\u8fdb\u884connx\u7684\u62bd\u53d6\u4e00\u5b9a\u62a5\u9519\uff0c\u5728\u4e0a\u4e00\u8282\u6211\u4eec\u5df2\u7ecf\u770b\u5230pose.pt\u6a21\u578b\u4f7f\u7528\u7684\u68c0\u6d4b\u5934\u4e3aIKeypoint\uff0c\u90a3\u4e48\u811a\u672c\u9700\u8981\u8fdb\u884c\u76f8\u5e94\u66f4\u6539\uff1a\u5728export.py\u7684\u8fd9\u4e2a\u4f4d\u7f6e\u63d2\u5165\uff1a<\/p>\n\n\n\n<pre class=\"wp-block-code\"><code> # \u539f\u4ee3\u7801:\n    for k, m in model.named_modules():\n        m._non_persistent_buffers_set = set()  # pytorch 1.6.0 compatibility\n        if isinstance(m, models.common.Conv):  # assign export-friendly activations\n            if isinstance(m.act, nn.Hardswish):\n                m.act = Hardswish()\n            elif isinstance(m.act, nn.SiLU):\n                m.act = SiLU()\n     model.model&#91;-1].export = not opt.grid  # set Detect() layer grid export\n                \n    # \u4fee\u6539\u4ee3\u7801:\n    for k, m in model.named_modules():\n        m._non_persistent_buffers_set = set()  # pytorch 1.6.0 compatibility\n        if isinstance(m, models.common.Conv):  # assign export-friendly activations\n            if isinstance(m.act, nn.Hardswish):\n                m.act = Hardswish()\n            elif isinstance(m.act, nn.SiLU):\n                m.act = SiLU()\n        elif isinstance(m, models.yolo.IKeypoint):\n            m.forward = m.forward_keypoint  # assign forward (optional)\n            # \u6b64\u5904\u5207\u6362\u68c0\u6d4b\u5934\n    model.model&#91;-1].export = not opt.grid  # set Detect() layer grid export<\/code><\/pre>\n\n\n\n<p>forward_keypoint\u5728\u539f\u59cb\u7684yolov7 repo\u6e90\u7801\u4e2d\u6709\uff0c\u4f5c\u8005\u5df2\u7ecf\u5c01\u88c5\u597d\uff0c\u4f46\u4f30\u8ba1\u662f\u8fd8\u6ca1\u6253\u7b97\u5f00\u653e\u4f7f\u7528\u3002<\/p>\n\n\n\n<p>\u4f7f\u7528\u4ee5\u4e0b\u547d\u4ee4\u8fdb\u884c\u62bd\u53d6\uff1apython&nbsp;export.py&nbsp;&#8211;weights&nbsp;&#8216;weights\/yolov7-w6-pose.pt&#8217;&nbsp;&#8211;img-size&nbsp;960&nbsp;&#8211;simplify&nbsp;True<\/p>\n\n\n\n<figure class=\"wp-block-image size-full\"><img loading=\"lazy\" width=\"702\" height=\"882\" src=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-18.png\" alt=\"\" class=\"wp-image-5166\" srcset=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-18.png 702w, http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-18-239x300.png 239w\" sizes=\"(max-width: 702px) 100vw, 702px\" \/><\/figure>\n\n\n\n<p><strong>\u4e09\u3001onnxruntime\u63a8\u7406<\/strong><\/p>\n\n\n\n<p>onnxruntime\u63a8\u7406\u4ee3\u7801\uff1a<\/p>\n\n\n\n<pre class=\"wp-block-preformatted\">import onnxruntime\nimport matplotlib.pyplot as plt\nimport torch\nimport cv2\nfrom torchvision import transforms\nimport numpy as np\nfrom utils.datasets import letterbox\nfrom utils.general import non_max_suppression_kpt\nfrom utils.plots import output_to_keypoint, plot_skeleton_kpts\n\ndevice = torch.device(\"cpu\")\n\nimage = cv2.imread('sample\/pose.jpeg')\nimage = letterbox(image, 960, stride=64, auto=True)[0]\nimage_ = image.copy()\nimage = transforms.ToTensor()(image)\nimage = torch.tensor(np.array([image.numpy()]))\n\nprint(image.shape)\nsess = onnxruntime.InferenceSession('weights\/yolov7-w6-pose.onnx')\nout = sess.run(['output'], {'images': image.numpy()})[0]\nout = torch.from_numpy(out)\n\noutput = non_max_suppression_kpt(out, 0.25, 0.65, nc=1, nkpt=17, kpt_label=True)\noutput = output_to_keypoint(output)\nnimg = image[0].permute(1, 2, 0) * 255\nnimg = nimg.cpu().numpy().astype(np.uint8)\nnimg = cv2.cvtColor(nimg, cv2.COLOR_RGB2BGR)\nfor idx in range(output.shape[0]):\n    plot_skeleton_kpts(nimg, output[idx, 7:].T, 3)\n\n# matplotlib inline\nplt.figure(figsize=(8, 8))\nplt.axis('off')\nplt.imshow(nimg)\nplt.show()\nplt.savefig(\"tmp\")<\/pre>\n\n\n\n<p class=\"has-normal-font-size\">\u63a8\u7406\u6548\u679c\u51e0\u4e4e\u65e0\u635f\uff0c\u4f46\u8017\u65f6\u4f1a\u7f29\u77ed\u4e00\u500d\u5de6\u53f3\uff0c\u53e6\u5916\u6709\u51e0\u4e2a\u70b9\uff1a<\/p>\n\n\n\n<ul><li>image = letterbox(image, 960, stride=64, auto=True)[0] \u4e2dstride\u6307\u7684\u662f\u6700\u5927\u6b65\u957f\uff0cyolov7-w6\u548cyolov5s\u4e0b\u91c7\u6837\u591a\u4e86\u4e00\u6b65\uff0c\u5bfc\u81f4\u57288\uff0c16\uff0c32\u7684\u57fa\u7840\u4e0a\u591a\u4e8664\u7684\u4e0b\u91c7\u6837\u6b65\u957f<\/li><li>output = non_max_suppression_kpt(out, 0.25, 0.65, nc=1, nkpt=17, kpt_label=True) \uff0cnc \u548c kpt_label \u7b49\u4fe1\u606f\u5728netron\u6253\u5370\u6a21\u578b\u6587\u4ef6\u65f6\u53ef\u4ee5\u770b\u5230<\/li><li>\u6240\u5f97\u5230\u7684onnx\u76f8\u6bd4\u539f\u534a\u7cbe\u5ea6\u6a21\u578b\u5927\u4e86\u5c06\u8fd1\u4e09\u500d\uff0c\u540e\u7eed\u6392\u67e5\u539f\u56e0<\/li><li>yolov7-w6-pose\u6781\u5ea6\u5403\u663e\u5b58\uff0c\u63a8\u7406\u4e00\u5f20960\u00d7960\u7684\u56fe\u50cf\uff0c\u9700\u89812-4G\u7684\u663e\u5b58\uff0c\u8bad\u7ec3\u66f4\u96be\u4ee5\u60f3\u8c61<\/li><\/ul>\n\n\n\n<figure class=\"wp-block-image size-full\"><img loading=\"lazy\" width=\"828\" height=\"556\" src=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-19.png\" alt=\"\" class=\"wp-image-5179\" srcset=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-19.png 828w, http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-19-300x201.png 300w, http:\/\/139.9.1.231\/wp-content\/uploads\/2022\/08\/image-19-768x516.png 768w\" sizes=\"(max-width: 828px) 100vw, 828px\" \/><\/figure>\n\n\n\n<p><\/p>\n","protected":false},"excerpt":{"rendered":"<p>\u76ee\u524d\u4eba\u4f53\u59ff\u6001\u4f30\u8ba1\u603b\u4f53\u5206\u4e3aTop-down\u548cBottom-up\u4e24\u79cd\uff0c\u4e0e\u76ee\u6807\u68c0\u6d4b\u4e0d\u540c\uff0c\u65e0\u8bba\u662f\u57fa\u4e8e\u70ed\u529b\u56fe\u6216\u662f\u57fa\u4e8e\u68c0\u6d4b &hellip; <a href=\"http:\/\/139.9.1.231\/index.php\/2022\/08\/02\/yolov7-pose\/\" class=\"more-link\">\u7ee7\u7eed\u9605\u8bfb<span class=\"screen-reader-text\">YOLOv7-Pose  \u57fa\u4e8eYOLOv7\u7684\u5173\u952e\u70b9\u6a21\u578b<\/span><\/a><\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":[],"categories":[4,9],"tags":[],"_links":{"self":[{"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/posts\/5147"}],"collection":[{"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/comments?post=5147"}],"version-history":[{"count":9,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/posts\/5147\/revisions"}],"predecessor-version":[{"id":5180,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/posts\/5147\/revisions\/5180"}],"wp:attachment":[{"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/media?parent=5147"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/categories?post=5147"},{"taxonomy":"post_tag","embeddable":true,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/tags?post=5147"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}