{"id":17133,"date":"2022-08-25T10:00:00","date_gmt":"2022-08-25T02:00:00","guid":{"rendered":"https:\/\/www.circuspi.com\/?p=17133"},"modified":"2022-08-25T10:41:53","modified_gmt":"2022-08-25T02:41:53","slug":"recomputer-jetson-yolov5-tensorrt","status":"publish","type":"post","link":"https:\/\/www.circuspi.com\/index.php\/2022\/08\/25\/recomputer-jetson-yolov5-tensorrt\/","title":{"rendered":"reComputer Jetson \u904b\u884c YOLOv5 \u6559\u5b78\u5168\u653b\u7565 3 &#8211; \u4f7f\u7528 TensorRT \u52a0\u901f\u63a8\u8ad6"},"content":{"rendered":"\n<p><style>\n\/* ###\u5be6\u9a57\u4e0b\u4e00\u5217### *\/\n\na:link {color:#0695e4\n;}    \/* \u8a2d\u5b9a\u5c1a\u672a\u9ede\u95b1\u904e\u7684\u9023\u7d50\u6a23\u5f0f *\/\na:visited {color:#0695e4\n;} \/* \u8a2d\u5b9a\u904e\u53bb\u66fe\u7d93\u95b1\u904e\u7684\u9023\u7d50 *\/\na:hover {color:#0695e4\n;}   \/* \u8a2d\u5b9a\u6ed1\u9f20\u6e38\u6a19\u6307\u5728\u9023\u7d50\u4f4d\u7f6e\u4e0a\u7684\u6a23\u5f0f *\/\na:active {color:#0000BB;}  \/* \u8a2d\u5b9a\u4e0a\u9ede\u64ca\u904e\u7684\u9023\u7d50\u6a23\u5f0f *\/\n<\/style><\/p>\n\n\n\n<figure class=\"wp-block-image size-full\"><img loading=\"lazy\" decoding=\"async\" width=\"1000\" height=\"526\" src=\"https:\/\/www.circuspi.com\/wp-content\/uploads\/2022\/08\/1000x526-5.jpg\" alt=\"\" class=\"wp-image-17269\"\/><\/figure>\n\n\n\n<h2 class=\"wp-block-heading\">\u4e00\u3001\u524d\u8a00<\/h2>\n\n\n\n<p>\u5982\u679c\u6709\u8ddf\u8457\u4e0a\u4e00\u7bc7\u6587\u7ae0\u64cd\u4f5c\u7684\u8a71\uff0c\u76f8\u4fe1\u5927\u5bb6\u5df2\u7d93\u8a13\u7df4\u597d\u81ea\u5df1\u7684\u6a21\u578b best.pt \u4e86\u3002\u9019\u6b21\u5c31\u662f\u8981\u6559\u5927\u5bb6\u5982\u4f55\u4f7f\u7528\u81ea\u5df1\u7684\u6a21\u578b\u505a\u63a8\u8ad6\uff0c\u4e26\u7d93\u7531TensorRT\uff0c\u91dd\u5c0d\u81ea\u5df1\u7684\u6a21\u578b\u8207\u786c\u9ad4\u505a\u52a0\u901f\u3002\u4e8b\u4e0d\u5b9c\u9072\uff0c\u8b93\u6211\u5011\u99ac\u4e0a\u958b\u59cb\uff01<\/p>\n\n\n\n<h2 class=\"wp-block-heading\">\u4e8c\u3001\u4f7f\u7528\u81ea\u5df1\u7684\u6a21\u578b<\/h2>\n\n\n\n<p><a href=\"https:\/\/pytorch.org\/hub\/ultralytics_yolov5\/\" target=\"_blank\" rel=\"noreferrer noopener\">YOLOv5<\/a> \u5b98\u65b9\u6587\u4ef6\u5176\u5be6\u63d0\u4f9b\u4e86\u5e7e\u500b\u65b9\u6cd5\uff0c\u8b93\u4f7f\u7528\u8005\u53ef\u4ee5\u4f7f\u7528\u81ea\u5df1\u7684\u6a21\u578b(Custom Model)\uff0c\u90a3\u5728\u9019\u908a\u6211\u5011\u5c31\u7528\u719f\u6089\u7684\u65b9\u5f0f\u300cdetect.py\u300d\u4f86\u57f7\u884c\u3002<\/p>\n\n\n\n<p>\u4e0a\u7bc7\u300c<a href=\"https:\/\/www.circuspi.com\/wp-admin\/post.php?post=17048&amp;action=edit\" target=\"_blank\" rel=\"noreferrer noopener\">reComputer Jetson<\/a> <a href=\"https:\/\/www.circuspi.com\/wp-admin\/post.php?post=17048&amp;action=edit\" target=\"_blank\" rel=\"noreferrer noopener\">\u904b\u884c YOLOv5 \u6559\u5b78\u5168\u653b\u7565 2 \u2013 \u8a13\u7df4\u81ea\u5df1\u7684\u6a21\u578b<\/a>\u300d\u6587\u7ae0\u57f7\u884c\u5b8c Colab \u7a0b\u5f0f\u78bc\u5f8c\uff0c\u8a13\u7df4\u5b8c\u6210\u7684\u6a21\u578b\u61c9\u8a72\u6703\u5132\u5b58\u5728\u300cDownloads\u300d\u8cc7\u6599\u593e\u5167\u3002<\/p>\n\n\n\n<div class=\"wp-block-image\"><figure class=\"aligncenter size-full\"><img loading=\"lazy\" decoding=\"async\" width=\"700\" height=\"401\" src=\"https:\/\/www.circuspi.com\/wp-content\/uploads\/2022\/08\/01-1.png\" alt=\"reComputer J1010 \u904b\u884c YOLOv5 \u6559\u5b78\u5168\u653b\u7565 3 - \u4f7f\u7528 TensorRT \u52a0\u901f\u63a8\u8ad6\" class=\"wp-image-17141\"\/><\/figure><\/div>\n\n\n\n<p>\u8acb\u5c07\u9019\u500b\u300cbest.pt\u300d\u6a21\u578b\uff0c\u8907\u88fd\u5230\u300cyolov5\u300d\u8cc7\u6599\u593e\u5167\u3002\u5f85\u6703\u5152\u6211\u5011\u5c07\u6703\u7528\u7a0b\u5f0f\uff0c\u547c\u53eb\u9019\u500b\u6a21\u578b\u3002<\/p>\n\n\n\n<div class=\"wp-block-image\"><figure class=\"aligncenter size-full\"><img loading=\"lazy\" decoding=\"async\" width=\"700\" height=\"388\" src=\"https:\/\/www.circuspi.com\/wp-content\/uploads\/2022\/08\/02-1.png\" alt=\"reComputer J1010 \u904b\u884c YOLOv5 \u6559\u5b78\u5168\u653b\u7565 3 - \u4f7f\u7528 TensorRT \u52a0\u901f\u63a8\u8ad6\" class=\"wp-image-17142\"\/><\/figure><\/div>\n\n\n\n<p>\u63a5\u8457\u6253\u958b\u7d42\u7aef\u6a5f\uff0c\u4e26\u8f38\u5165\u4e0b\u65b9\u6307\u4ee4\u9032\u5230 yolov5 \u8cc7\u6599\u593e\u5167\u3002<\/p>\n\n\n\n<pre><span style=\"color: #ff0000;\">cd<\/span> yolov5<\/pre>\n\n\n\n<div class=\"wp-block-image\"><figure class=\"aligncenter size-full\"><img loading=\"lazy\" decoding=\"async\" width=\"700\" height=\"413\" src=\"https:\/\/www.circuspi.com\/wp-content\/uploads\/2022\/08\/03-1.png\" alt=\"reComputer J1010 \u904b\u884c YOLOv5 \u6559\u5b78\u5168\u653b\u7565 3 - \u4f7f\u7528 TensorRT \u52a0\u901f\u63a8\u8ad6\" class=\"wp-image-17143\"\/><\/figure><\/div>\n\n\n\n<p>\u8207\u4e4b\u524d\u5728<span style=\"background-color:#fcb900\" class=\"tadv-background-color\">\u7b2c\u4e00\u7bc7\u6587\u7ae0<\/span>\u6e2c\u8a66 YOLOv5 \u6a21\u578b\u4e00\u6a23\uff0c\u6211\u5011\u6703\u547c\u53eb\u300cdetect.py\u300d\u7a0b\u5f0f\uff0c\u8acb\u8f38\u5165\u4e0b\u65b9\u6307\u4ee4\u547c\u53eb\u6211\u5011\u81ea\u5df1\u7684\u6a21\u578b\u505a\u63a8\u8ad6\u3002\u6bd4\u5c0d\u4e4b\u524d\u7684\u6307\u4ee4\u53ef\u4ee5\u767c\u73fe\uff0c\u53ea\u662f\u66ff\u63db\u6389\u300c&#8211;weights\u300d\u5f8c\u9762\u7684\u6a21\u578b\u540d\u7a31\u800c\u5df2\uff0c\u662f\u4e0d\u662f\u8d85\u7c21\u55ae\u3002<\/p>\n\n\n\n<pre>python3 detect.py --weights best.pt --source <span style=\"color: #a600ff;\">0<\/span> --nosave<\/pre>\n\n\n\n<div class=\"wp-block-image\"><figure class=\"aligncenter size-full\"><img loading=\"lazy\" decoding=\"async\" width=\"700\" height=\"410\" src=\"https:\/\/www.circuspi.com\/wp-content\/uploads\/2022\/08\/04-1.png\" alt=\"reComputer J1010 \u904b\u884c YOLOv5 \u6559\u5b78\u5168\u653b\u7565 3 - \u4f7f\u7528 TensorRT \u52a0\u901f\u63a8\u8ad6\" class=\"wp-image-17144\"\/><\/figure><\/div>\n\n\n\n<p>\u7b49\u5f85\u7a0b\u5f0f\u57f7\u884c\u5f8c\uff0c\u5c31\u80fd\u770b\u5230\u986f\u793a\u7d50\u679c\u8207\u63a8\u8ad6\u8cc7\u8a0a\u3002\u8d95\u7dca\u5c07\u81ea\u5df1\u60f3\u8981\u8fa8\u8b58\u7684\u7269\u9ad4\u653e\u5230\u651d\u5f71\u6a5f\u524d\uff0c\u770b\u770b\u6a21\u578b\u662f\u5426\u80fd\u6b63\u78ba\u8fa8\u8b58\u3002<\/p>\n\n\n\n<div class=\"wp-block-image\"><figure class=\"aligncenter size-full\"><img loading=\"lazy\" decoding=\"async\" width=\"700\" height=\"283\" src=\"https:\/\/www.circuspi.com\/wp-content\/uploads\/2022\/08\/05-1.png\" alt=\"reComputer J1010 \u904b\u884c YOLOv5 \u6559\u5b78\u5168\u653b\u7565 3 - \u4f7f\u7528 TensorRT \u52a0\u901f\u63a8\u8ad6\" class=\"wp-image-17145\"\/><\/figure><\/div>\n\n\n\n<p>\u67e5\u770b\u7d42\u7aef\u6a5f\u5167\u7684\u8cc7\u8a0a\uff0c\u53ef\u4ee5\u770b\u5230\u6bcf\u63a8\u8ad6\u4e00\u5e40\u5f71\u50cf\uff0c\u6703\u9700\u8981\u82b1\u8cbb\u7d04 61ms \u7684\u6642\u9593\u3002\u90a3\u6709\u6c92\u6709\u6a5f\u6703\u52a0\u901f\u63a8\u8ad6\u6642\u9593\u5462\uff1f\u6709\u7684\uff0c\u6211\u5011\u53ef\u4ee5\u501f\u52a9 <a href=\"https:\/\/developer.nvidia.com\/tensorrt\" target=\"_blank\" rel=\"noreferrer noopener\">TensorRT<\/a>\u3002<\/p>\n\n\n\n<h2 class=\"wp-block-heading\">\u4e09\u3001\u4ec0\u9ebc\u662f TensorRT ?<\/h2>\n\n\n\n<p><a href=\"https:\/\/developer.nvidia.com\/tensorrt\" target=\"_blank\" rel=\"noreferrer noopener\">TensorRT<\/a> \u662f <a href=\"https:\/\/www.nvidia.com\/zh-tw\/\" target=\"_blank\" rel=\"noreferrer noopener\">NVIDIA<\/a> \u91dd\u5c0d\u81ea\u5df1\u7684\u7522\u54c1\u6240\u505a\u7684\u4e00\u500b\u5de5\u5177\uff0c\u4e3b\u8981\u76ee\u7684\u5728\u65bc\u5354\u52a9\u4f7f\u7528\u8005\u7684\u6a21\u578b\uff0c\u80fd\u66f4\u5feb\u66f4\u6709\u6548\u7684\u53d6\u7528\u81ea\u5bb6 GPU \u8cc7\u6e90\uff0c\u4e26\u900f\u904e\u4e00\u4e9b\u512a\u5316\u7a0b\u5e8f\uff0c\u52a0\u901f\u6a21\u578b\u7684\u63a8\u8ad6\u3002\u7531\u4e0b\u5716\u53ef\u4ee5\u770b\u5230\uff0c\u5b83\u652f\u63f4\u73fe\u4eca\u5e38\u898b\u4e14\u6d41\u884c\u7684\u795e\u7d93\u7db2\u8def\u6846\u67b6\u3002<\/p>\n\n\n\n<div class=\"wp-block-image\"><figure class=\"aligncenter size-full\"><img loading=\"lazy\" decoding=\"async\" width=\"700\" height=\"309\" src=\"https:\/\/www.circuspi.com\/wp-content\/uploads\/2022\/08\/06-1.png\" alt=\"reComputer J1010 \u904b\u884c YOLOv5 \u6559\u5b78\u5168\u653b\u7565 3 - \u4f7f\u7528 TensorRT \u52a0\u901f\u63a8\u8ad6\" class=\"wp-image-17146\"\/><figcaption>\u5716\u7247\u4f86\u6e90\uff1a\u53d6\u81ea <a href=\"https:\/\/www.nvidia.com\/zh-tw\/\" target=\"_blank\" rel=\"noreferrer noopener\">NVIDIA<\/a> \u5b98\u65b9\u7db2\u7ad9<\/figcaption><\/figure><\/div>\n\n\n\n<p><a href=\"https:\/\/developer.nvidia.com\/tensorrt\" target=\"_blank\" rel=\"noreferrer noopener\">TensorRT<\/a> \u512a\u5316\u7684\u9805\u76ee\u5982\u4e0b\u5716\u6240\u793a\uff0c\u5171\u6709\u516d\u9805\u3002\u900f\u904e\u8f49\u63db\u6a21\u578b\u964d\u4f4e\u7cbe\u5ea6\uff0c\u4ee5\u53ca\u512a\u5316\u6a21\u578b\u7d50\u69cb\uff0c\u4e26\u91dd\u5c0d\u4f7f\u7528\u7684\u786c\u9ad4\u505a\u512a\u5316\u3002\u9019\u88e1\u6709\u500b\u7279\u5225\u7684\u91cd\u9ede\uff0c\u5c31\u662f\u91dd\u5c0d\u786c\u9ad4\u512a\u5316\u9019\u4ef6\u4e8b\uff0c\u4e5f\u5c31\u662f\u8aaa\u5927\u5bb6\u82e5\u662f\u5728 Jetson Nano \u4e0a\uff0c\u57f7\u884c\u4e86 TensorRT \u512a\u5316\u6a21\u578b\uff0c\u90a3\u9ebc\u7522\u751f\u51fa\u4f86\u7684\u6a21\u578b\u6a94\u6848\u300c\u6a21\u578b\u540d\u7a31.engine\u300d\uff0c\u5c31\u4e0d\u80fd\u8907\u88fd\u5230\u5176\u4ed6\u4e0d\u540c\u7684\u786c\u9ad4\u505a\u4f7f\u7528\u3002<\/p>\n\n\n\n<div class=\"wp-block-image\"><figure class=\"aligncenter size-full\"><img loading=\"lazy\" decoding=\"async\" width=\"700\" height=\"217\" src=\"https:\/\/www.circuspi.com\/wp-content\/uploads\/2022\/08\/07-1.png\" alt=\"reComputer J1010 \u904b\u884c YOLOv5 \u6559\u5b78\u5168\u653b\u7565 3 - \u4f7f\u7528 TensorRT \u52a0\u901f\u63a8\u8ad6\" class=\"wp-image-17147\"\/><figcaption>\u5716\u7247\u53d6\u81ea <a href=\"https:\/\/www.nvidia.com\/zh-tw\/\" target=\"_blank\" rel=\"noreferrer noopener\">NVIDIA<\/a> \u5b98\u65b9\u7db2\u7ad9<\/figcaption><\/figure><\/div>\n\n\n\n<h2 class=\"wp-block-heading\">\u56db\u3001\u4f7f\u7528 TensorRT<\/h2>\n\n\n\n<p>\u77e5\u9053 TensorRT \u7684\u597d\u8655\u5f8c\uff0c\u7576\u7136\u8981\u76f4\u63a5\u5c07\u6211\u5011\u7684\u6a21\u578b\u8f49\u63db\u4e00\u6ce2\u5440\uff01\u8f49\u63db\u7684\u65b9\u6cd5\u4e5f\u76f8\u7576\u7c21\u55ae\uff0cYOLOv5 \u5b98\u65b9\u7a0b\u5f0f\u78bc\u6709\u63d0\u4f9b\u8f49\u63db\u7a0b\u5f0f\u300cexport.py\u300d\uff0c\u9019\u500b\u7a0b\u5f0f\u63d0\u4f9b\u4f7f\u7528\u8005\u80fd\u5c07\u81ea\u5df1\u7684 YOLOv5 \u6a21\u578b\u8f49\u63db\u6210\u4e0d\u540c\u7684\u683c\u5f0f\uff0c\u5176\u4e2d\u5c31\u5305\u62ec TensorRT \u7684\u300c.engine\u300d\u6a94\u6848\u3002<\/p>\n\n\n\n<p>\u4f46\u5728\u8f49\u63db\u6210 .engine \u6a94\u6848\u7684\u904e\u7a0b\u4e2d\uff0c\u6703\u9700\u8981\u5148\u5c07 Pytorch \u7684\u300c.pt\u300d\u8f49\u63db\u6210\u300c.onnx\u300d\uff0c\u518d\u7531\u300c.onnx\u300d\u8f49\u63db\u6210\u6211\u5011\u8981\u7684\u300c.engine\u300d\u6a94\u6848\u3002<\/p>\n\n\n\n<p>onnx(Open Neural Network Exchange)\u662f\u4e00\u5957\u958b\u653e\u795e\u7d93\u7db2\u8def\u4ea4\u63db\u683c\u5f0f\uff0c\u5728\u5b89\u88dd YOLOv5 \u7684\u6642\u5019\u4e26\u6c92\u6709\u5b89\u88dd\u76f8\u95dc\u8edf\u9ad4\uff0c\u6240\u4ee5\u5728\u8f49\u63db\u524d\uff0c\u6211\u5011\u9700\u8981\u5148\u624b\u52d5\u5b89\u88dd onnx\u3002<\/p>\n\n\n\n<h3 class=\"wp-block-heading\">1. \u5b89\u88dd onnx<\/h3>\n\n\n\n<p>\u9996\u5148\u8981\u5b89\u88dd\u76f8\u4f9d\u5957\u4ef6\uff0c\u8acb\u6253\u958b\u7d42\u7aef\u6a5f\u4e26\u8f38\u5165\u4e0b\u5217\u6307\u4ee4\u3002<\/p>\n\n\n\n<pre><span style=\"color: #ff0000;\">sudo apt install<\/span> protobuf-compiler libprotoc-dev<\/pre>\n\n\n\n<div class=\"wp-block-image\"><figure class=\"aligncenter size-full\"><img loading=\"lazy\" decoding=\"async\" width=\"700\" height=\"439\" src=\"https:\/\/www.circuspi.com\/wp-content\/uploads\/2022\/08\/08-1.png\" alt=\"reComputer J1010 \u904b\u884c YOLOv5 \u6559\u5b78\u5168\u653b\u7565 3 - \u4f7f\u7528 TensorRT \u52a0\u901f\u63a8\u8ad6\" class=\"wp-image-17148\"\/><\/figure><\/div>\n\n\n\n<p>\u904e\u7a0b\u4e2d\u6703\u9700\u8981\u4f7f\u7528\u8005\u540c\u610f\uff0c\u8acb\u8f38\u5165\u300cy\u300d\u4e26\u6309\u4e0b Enter \u540c\u610f\u5b89\u88dd\u3002<\/p>\n\n\n\n<div class=\"wp-block-image\"><figure class=\"aligncenter size-full\"><img loading=\"lazy\" decoding=\"async\" width=\"700\" height=\"429\" src=\"https:\/\/www.circuspi.com\/wp-content\/uploads\/2022\/08\/09-1.png\" alt=\"reComputer J1010 \u904b\u884c YOLOv5 \u6559\u5b78\u5168\u653b\u7565 3 - \u4f7f\u7528 TensorRT \u52a0\u901f\u63a8\u8ad6\" class=\"wp-image-17149\"\/><\/figure><\/div>\n\n\n\n<p>\u63a5\u8457\u8acb\u8f38\u5165\u4e0b\u5217\u6307\u4ee4\u5b89\u88ddCython\u3002<\/p>\n\n\n\n<pre>pip3 <span style=\"color: #ff0000;\">install<\/span> Cython --user<\/pre>\n\n\n\n<div class=\"wp-block-image\"><figure class=\"aligncenter size-full\"><img loading=\"lazy\" decoding=\"async\" width=\"700\" height=\"442\" src=\"https:\/\/www.circuspi.com\/wp-content\/uploads\/2022\/08\/10-1.png\" alt=\"reComputer J1010 \u904b\u884c YOLOv5 \u6559\u5b78\u5168\u653b\u7565 3 - \u4f7f\u7528 TensorRT \u52a0\u901f\u63a8\u8ad6\" class=\"wp-image-17150\"\/><\/figure><\/div>\n\n\n\n<p>\u6700\u5f8c\u5b89\u88dd onnx\uff0c\u9019\u908a\u9700\u8981\u6307\u5b9a\u7248\u672c\uff0c\u8acb\u8f38\u5165\u4e0b\u5217\u6307\u4ee4\u5b89\u88dd\u3002<\/p>\n\n\n\n<pre>pip3 <span style=\"color: #ff0000;\">install<\/span> <span style=\"color: #e37932;\">onnx<\/span>==<span style=\"color: #ff00ff;\">1.9.0<\/span> --user --verbose<\/pre>\n\n\n\n<div class=\"wp-block-image\"><figure class=\"aligncenter size-full\"><img loading=\"lazy\" decoding=\"async\" width=\"700\" height=\"451\" src=\"https:\/\/www.circuspi.com\/wp-content\/uploads\/2022\/08\/11-1.png\" alt=\"reComputer J1010 \u904b\u884c YOLOv5 \u6559\u5b78\u5168\u653b\u7565 3 - \u4f7f\u7528 TensorRT \u52a0\u901f\u63a8\u8ad6\" class=\"wp-image-17151\"\/><\/figure><\/div>\n\n\n\n<p>\u5b89\u88dd\u5b8c\u6210\u5f8c\u5982\u4e0b\u6240\u793a\uff1a<\/p>\n\n\n\n<div class=\"wp-block-image\"><figure class=\"aligncenter size-full\"><img loading=\"lazy\" decoding=\"async\" width=\"700\" height=\"441\" src=\"https:\/\/www.circuspi.com\/wp-content\/uploads\/2022\/08\/12-1.png\" alt=\"reComputer J1010 \u904b\u884c YOLOv5 \u6559\u5b78\u5168\u653b\u7565 3 - \u4f7f\u7528 TensorRT \u52a0\u901f\u63a8\u8ad6\" class=\"wp-image-17152\"\/><\/figure><\/div>\n\n\n\n<h3 class=\"wp-block-heading\">2. \u4f7f\u7528 TensorRT \u512a\u5316\u6a21\u578b<\/h3>\n\n\n\n<p>\u8f49\u63db\u6a21\u578b\u4e4b\u524d\u6211\u5011\u9700\u8981\u65b0\u589e\u300c\u6a19\u7c64\u300d\u5c0d\u61c9\u6a94\u6848\uff0c\u9644\u6a94\u540d\u662f\u300cyaml\u300d\uff0c\u9019\u500b\u6a94\u6848\u6e05\u695a\u5beb\u660e\u6211\u5011\u9019\u500b\u6a21\u578b\u8fa8\u8b58\u7684\u985e\u5225\u6578\u91cf\uff0c\u8207\u6bcf\u500b\u8fa8\u8b58\u5230\u7684\u985e\u5225\u540d\u7a31\u662f\u4ec0\u9ebc\u3002<\/p>\n\n\n\n<p>\u8acb\u5728 yolov5 \u8cc7\u6599\u593e\u5e95\u4e0b\uff0c\u8f38\u5165\u4e0b\u5217\u6307\u4ee4\u958b\u555f\u6587\u5b57\u7de8\u8f2f\u5668\uff0c\u65b0\u589e\u300clabels.yaml\u300d\u6a94\u6848\uff1a<\/p>\n\n\n\n<pre>gedit labels.yaml<\/pre>\n\n\n\n<div class=\"wp-block-image\"><figure class=\"aligncenter size-full\"><img loading=\"lazy\" decoding=\"async\" width=\"700\" height=\"426\" src=\"https:\/\/www.circuspi.com\/wp-content\/uploads\/2022\/08\/13-1.png\" alt=\"reComputer J1010 \u904b\u884c YOLOv5 \u6559\u5b78\u5168\u653b\u7565 3 - \u4f7f\u7528 TensorRT \u52a0\u901f\u63a8\u8ad6\" class=\"wp-image-17153\"\/><\/figure><\/div>\n\n\n\n<p>\u8acb\u53c3\u7167\u4e0b\u5217\u6587\u4ef6\u683c\u5f0f\uff0c\u4f9d\u7167\u81ea\u5df1\u7684\u6a21\u578b\u4fee\u6539\u5167\u5bb9\u3002<\/p>\n\n\n\n<pre><span style=\"color: #0000ff;\"># Classes<\/span><br><span style=\"color: #993300;\">nc<\/span>: <span style=\"color: #ff00ff;\">2<\/span>&nbsp; <span style=\"color: #0000ff;\"># number of classes<\/span><br><span style=\"color: #993300;\">names<\/span>: [<span style=\"color: #ff00ff;\">'hole', 'radish'<\/span>]<\/pre>\n\n\n\n<div class=\"wp-block-image\"><figure class=\"aligncenter size-full\"><img loading=\"lazy\" decoding=\"async\" width=\"700\" height=\"412\" src=\"https:\/\/www.circuspi.com\/wp-content\/uploads\/2022\/08\/14-2.png\" alt=\"reComputer J1010 \u904b\u884c YOLOv5 \u6559\u5b78\u5168\u653b\u7565 3 - \u4f7f\u7528 TensorRT \u52a0\u901f\u63a8\u8ad6\" class=\"wp-image-17155\"\/><\/figure><\/div>\n\n\n\n<p>\u82e5\u60a8\u7684\u8fa8\u8b58\u6578\u91cf\u662f 3 \u7a2e\uff0c\u8acb\u4fee\u6539\u300cnc: \u300d\u5f8c\u65b9\u7684\u6578\u5b57\uff0c\u4f8b\u5982\uff1a\u300cnc: 3\u300d\u3002\u7136\u5f8c\u5728\u300cnames: \u300d\u5f8c\u65b9\u7684\u9663\u5217\u4e2d\u65b0\u589e\u81ea\u5df1\u7684\u6a19\u7c64\u540d\u7a31\uff0c\u6bd4\u65b9\u8aaa\u60a8\u7684\u6a21\u578b\u53ef\u4ee5\u8fa8\u8b58\u5230\u4e09\u7a2e\u6c34\u679c\uff0c\u8acb\u5728\u9663\u5217\u7576\u4e2d\u653e\u5165\u500b\u6c34\u679c\u540d\u7a31\u7684\u5b57\u4e32\uff0c\u4e26\u7528\u9017\u865f\u9694\u958b\uff0c\u4f8b\u5982\uff1a\u300cnames: [\u2018strawberry\u2019, \u2018orange\u2019, \u2018banana\u2019]\u300d\u3002\u4ee5\u6b64\u985e\u63a8\uff0c\u4fee\u6539\u5b8c\u6210\u5f8c\u8acb\u5b58\u6a94\uff0c\u4e26\u95dc\u9589\u8996\u7a97\u3002<\/p>\n\n\n\n<p>OK~\u4e00\u5207\u6e96\u5099\u5c31\u7dd2\uff0c\u6a21\u578b\u6709\u4e86\u3001\u6a19\u7c64\u6a94\u6848\u6709\u4e86\u4ee5\u53ca onnx \u4e5f\u5b89\u88dd\u4e86\uff0c\u63a5\u4e0b\u4f86\u8b93\u6211\u5011\u8f38\u5165\u4ee5\u4e0b\u6307\u4ee4\uff0c\u958b\u59cb\u8f49\u63db\u6211\u5011\u7684\u6a21\u578b\u3002<\/p>\n\n\n\n<pre>python3 export.py --weights best.pt --data labels.yaml --include engine --half --device 0<\/pre>\n\n\n\n<div class=\"wp-block-image\"><figure class=\"aligncenter size-full\"><img loading=\"lazy\" decoding=\"async\" width=\"700\" height=\"431\" src=\"https:\/\/www.circuspi.com\/wp-content\/uploads\/2022\/08\/15-1.png\" alt=\"reComputer J1010 \u904b\u884c YOLOv5 \u6559\u5b78\u5168\u653b\u7565 3 - \u4f7f\u7528 TensorRT \u52a0\u901f\u63a8\u8ad6\" class=\"wp-image-17156\"\/><\/figure><\/div>\n\n\n\n<p>\u6307\u4ee4\u7576\u4e2d\u7684\u300c&#8211;data\u300d\u5f8c\u65b9\u5c31\u662f\u586b\u5165\u6211\u5011\u7684\u6a19\u7c64\u6a94\u6848\uff0c\u300c&#8211;include\u300d\u5f8c\u65b9\u5247\u662f\u586b\u5165\u8981\u8f49\u63db\u7684\u6a21\u578b\u985e\u578b\uff0c\u56e0\u70ba\u8981\u8f49\u63db\u7684\u662f TensorRT\uff0c\u6240\u4ee5\u662f\u300cengine\u300d\u3002\u300c&#8211;half\u300d\u5247\u662f\u6211\u5011\u8981\u5c07\u539f\u5148\u7684\u8fa8\u8b58\u7cbe\u5ea6 FP32 \u4e0b\u8abf\u6210 FP16\uff0c\u4ee5\u4e0b\u4fee\u7cbe\u5ea6\u63db\u5f97\u901f\u5ea6\u7684\u65b9\u5f0f\u52a0\u5feb\u63a8\u8ad6\u6642\u9593\u3002\u6700\u5f8c\u7684\u300c&#8211;device\u300d\u5f8c\u65b9\u5247\u662f\u586b\u5165\u4f7f\u7528\u7684 GPU \u7de8\u865f\uff0c\u5b83\u6703\u4f9d\u7167\u4f7f\u7528\u7684 GPU \u505a\u786c\u9ad4\u512a\u5316\uff0cJetson Nano \u53ea\u6709\u4e00\u500b GPU\uff0c\u7de8\u865f\u662f 0\u3002<\/p>\n\n\n\n<p>\u8f49\u63db\u904e\u7a0b\u9700\u8981\u5f88\u9577\u4e00\u6bb5\u6642\u9593\uff0c\u8f49\u63db\u5b8c\u7562\u5f8c\u5982\u4e0b\u5716\u6240\u793a\uff0c\u5167\u5bb9\u986f\u793a\u7b46\u8005\u9019\u6b21\u8f49\u63db\u6642\u9593\u9700\u8981 613.46 \u79d2\uff0c\u5c07\u8fd1 10 \u5206\u9418\u591a\u3002\u4ee5\u53ca\u63d0\u4f9b\u76f8\u95dc\u8cc7\u8a0a\uff0c\u4f8b\u5982\u8f49\u63db\u5f8c\u7684\u6a94\u6848\u653e\u5728\u54ea\uff0c\u8a72\u600e\u9ebc\u4f7f\u7528\u6a21\u578b&#8230;&#8230;\u7b49\u3002<\/p>\n\n\n\n<div class=\"wp-block-image\"><figure class=\"aligncenter size-full\"><img loading=\"lazy\" decoding=\"async\" width=\"700\" height=\"437\" src=\"https:\/\/www.circuspi.com\/wp-content\/uploads\/2022\/08\/16-1.png\" alt=\"reComputer J1010 \u904b\u884c YOLOv5 \u6559\u5b78\u5168\u653b\u7565 3 - \u4f7f\u7528 TensorRT \u52a0\u901f\u63a8\u8ad6\" class=\"wp-image-17157\"\/><\/figure><\/div>\n\n\n\n<p>\u5be6\u969b\u4f86\u5230 yolov5 \u8cc7\u6599\u593e\uff0c\u53ef\u4ee5\u627e\u5230\u8f49\u63db\u5f8c\u7684 engine \u6a94\u6848\u300cbest.engine\u300d\uff0c\u4ee5\u53ca\u8f49\u63db\u904e\u7a0b\u7522\u51fa\u7684 onnx \u6a94\u6848\u300cbest.onnx\u300d\u3002<\/p>\n\n\n\n<div class=\"wp-block-image\"><figure class=\"aligncenter size-full\"><img loading=\"lazy\" decoding=\"async\" width=\"700\" height=\"391\" src=\"https:\/\/www.circuspi.com\/wp-content\/uploads\/2022\/08\/17-1.png\" alt=\"reComputer J1010 \u904b\u884c YOLOv5 \u6559\u5b78\u5168\u653b\u7565 3 - \u4f7f\u7528 TensorRT \u52a0\u901f\u63a8\u8ad6\" class=\"wp-image-17158\"\/><\/figure><\/div>\n\n\n\n<h2 class=\"wp-block-heading\">\u4e94\u3001\u512a\u5316\u5f8c\u7684\u63a8\u8ad6\u7d50\u679c<\/h2>\n\n\n\n<p>\u7372\u5f97 engine \u6a94\u6848\u5f8c\uff0c\u99ac\u4e0a\u4f86\u6e2c\u8a66\u4e00\u4e0b\u6548\u679c\u5982\u4f55\u5427\uff01\u8acb\u8f38\u5165\u4ee5\u4e0b\u6307\u4ee4\uff1a<\/p>\n\n\n\n<pre>python3 detect.py --weights best.engine --half --data labels.yaml --nosave --source 0<\/pre>\n\n\n\n<div class=\"wp-block-image\"><figure class=\"aligncenter size-full\"><img loading=\"lazy\" decoding=\"async\" width=\"700\" height=\"428\" src=\"https:\/\/www.circuspi.com\/wp-content\/uploads\/2022\/08\/18-1.png\" alt=\"reComputer J1010 \u904b\u884c YOLOv5 \u6559\u5b78\u5168\u653b\u7565 3 - \u4f7f\u7528 TensorRT \u52a0\u901f\u63a8\u8ad6\" class=\"wp-image-17159\"\/><\/figure><\/div>\n\n\n\n<p>\u4e00\u6a23\u5c07\u300c&#8211;weights\u300d\u5f8c\u65b9\u66f4\u6539\u70ba\u6211\u5011\u7684 engine \u6a94\u6848\u300cbest.engine\u300d\uff0c\u4e26\u52a0\u5165\u300c&#8211;half\u300d\u8868\u793a\u6a21\u578b\u7cbe\u5ea6\u70ba FP16\uff0c\u4ee5\u53ca\u65b0\u589e\u300c&#8211;data\u300d\u8f38\u5165\u6a19\u7c64\u6a94\u6848\u300clabels.yaml\u300d\u3002<\/p>\n\n\n\n<p>\u5be6\u969b\u57f7\u884c\u5f8c\uff0c\u53ef\u4ee5\u770b\u5230\u63a8\u8ad6\u901f\u5ea6\u5927\u5e45\u63d0\u5347\uff0c\u8655\u7406\u6bcf\u5e40\u7684\u6642\u9593\u4e0b\u964d\u5230\u7d04 38ms\uff0c\u662f\u4e0d\u662f\u76f8\u7576\u9a5a\u4eba\uff01<\/p>\n\n\n\n<div class=\"wp-block-image\"><figure class=\"aligncenter size-full\"><img loading=\"lazy\" decoding=\"async\" width=\"700\" height=\"267\" src=\"https:\/\/www.circuspi.com\/wp-content\/uploads\/2022\/08\/19-1.png\" alt=\"reComputer J1010 \u904b\u884c YOLOv5 \u6559\u5b78\u5168\u653b\u7565 3 - \u4f7f\u7528 TensorRT \u52a0\u901f\u63a8\u8ad6\" class=\"wp-image-17160\"\/><\/figure><\/div>\n\n\n\n<h2 class=\"wp-block-heading\">\u516d\u3001\u7d50\u8a9e<\/h2>\n\n\n\n<p>\u7d9c\u5408\u4ee5\u4e0a\uff0c\u904b\u884c YOLOv5 \u5168\u653b\u7565\u7cfb\u5217\u5c31\u7b97\u5b8c\u7d50\u4e86\uff0c\u5f9e\u7b2c\u4e00\u7bc7\u6587\u7ae0\u7684\u74b0\u5883\u5b89\u88dd\uff0c\u7b2c\u4e8c\u7bc7\u8a13\u7df4\u81ea\u5df1\u7684\u6a21\u578b\uff0c\u5230\u9019\u7bc7\u7684\u4f7f\u7528 TensorRT \u52a0\u901f\uff0c\u76f8\u4fe1\u5927\u5bb6\u5df2\u7d93\u5c0d\u5728 <a href=\"https:\/\/www.icshop.com.tw\/product_list.php?2163\" target=\"_blank\" rel=\"noreferrer noopener\">Jetson Nano<\/a> \u4e0a\u9762\uff0c\u4f7f\u7528 YOLOv5 \u505a\u7269\u4ef6\u8fa8\u8b58\u6709\u6240\u4e86\u89e3\u3002\u82e5\u60a8\u624b\u908a\u9084\u6709\u66f4\u5feb\u7684 Jetson Xavier NX\u3001reComputer J2011\u3001reComputer J2012 \u6216\u662f\u6700\u65b0\u7684 Jetson AGX Orin\uff0c\u90fd\u6b61\u8fce\u60a8\u8a66\u8457\u5728\u4e0a\u9762\u57f7\u884c\u6211\u5011\u9019\u4e09\u7bc7\u7684\u5167\u5bb9\u770b\u770b\uff0c\u6216\u662f\u5c0b\u627e\u7db2\u8def\u4e0a\u5176\u4ed6\u6a21\u578b\uff0c\u8ddf\u8457\u5176\u4ed6\u5927\u795e\u7684\u8173\u6b65\u5b78\u7fd2\u3002\u82e5\u9019\u4e09\u7bc7\u7684\u5167\u5bb9\u6709\u932f\u8aa4\u6216\u662f\u63cf\u8ff0\u4e0d\u597d\u7684\u5730\u65b9\uff0c\u6b61\u8fce\u5927\u5bb6\u6307\u6b63\uff0c\u795d\u5927\u5bb6\u64cd\u4f5c\u9806\u5229\uff0c\u5b78\u7fd2\u6109\u5feb\uff01<\/p>\n","protected":false},"excerpt":{"rendered":"<p>\u4e0a\u4e00\u7bc7\u6587\u7ae0\u5df2\u7d93\u8a13\u7df4\u597d\u81ea\u5df1\u7684\u6a21\u578b best.pt \u4e86\uff0c\u9019\u6b21\u5c31\u662f\u8981\u6559\u5927\u5bb6\u5982\u4f55\u4f7f\u7528\u81ea\u5df1\u7684\u6a21\u578b\u505a\u63a8\u8ad6\uff0c\u4e26\u7d93\u7531 NVIDIA \u5b98\u65b9\u7684 TensorRT\uff0c\u91dd\u5c0d\u81ea\u5df1\u7684\u6a21\u578b\u8207\u786c\u9ad4\u505a\u52a0\u901f\u8207\u512a\u5316\uff0c\u5176\u4e2d\u6703\u6559\u5927\u5bb6\u4ec0\u9ebc\u662f TensorRT ?\u4ee5\u53ca\u5982\u4f55\u5b89\u88dd onnx\uff0c\u4e26\u5be6\u969b\u4f7f\u7528\u5b83\uff01<\/p>\n","protected":false},"author":1,"featured_media":17270,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[286,238],"tags":[42,119,290],"table_tags":[],"_links":{"self":[{"href":"https:\/\/www.circuspi.com\/index.php\/wp-json\/wp\/v2\/posts\/17133"}],"collection":[{"href":"https:\/\/www.circuspi.com\/index.php\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/www.circuspi.com\/index.php\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/www.circuspi.com\/index.php\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/www.circuspi.com\/index.php\/wp-json\/wp\/v2\/comments?post=17133"}],"version-history":[{"count":13,"href":"https:\/\/www.circuspi.com\/index.php\/wp-json\/wp\/v2\/posts\/17133\/revisions"}],"predecessor-version":[{"id":17271,"href":"https:\/\/www.circuspi.com\/index.php\/wp-json\/wp\/v2\/posts\/17133\/revisions\/17271"}],"wp:featuredmedia":[{"embeddable":true,"href":"https:\/\/www.circuspi.com\/index.php\/wp-json\/wp\/v2\/media\/17270"}],"wp:attachment":[{"href":"https:\/\/www.circuspi.com\/index.php\/wp-json\/wp\/v2\/media?parent=17133"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/www.circuspi.com\/index.php\/wp-json\/wp\/v2\/categories?post=17133"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/www.circuspi.com\/index.php\/wp-json\/wp\/v2\/tags?post=17133"},{"taxonomy":"table_tags","embeddable":true,"href":"https:\/\/www.circuspi.com\/index.php\/wp-json\/wp\/v2\/table_tags?post=17133"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}