{"id":13436244,"url":"https://github.com/spacewalk01/tensorrt-yolov9","last_synced_at":"2025-03-18T20:31:11.213Z","repository":{"id":223993650,"uuid":"762107819","full_name":"spacewalk01/tensorrt-yolov9","owner":"spacewalk01","description":"Cpp and python implementation of YOLOv9 using TensorRT API","archived":false,"fork":false,"pushed_at":"2024-09-30T02:31:05.000Z","size":16683,"stargazers_count":110,"open_issues_count":2,"forks_count":18,"subscribers_count":2,"default_branch":"main","last_synced_at":"2024-10-27T20:18:54.581Z","etag":null,"topics":["cpp","object-detection","tensorrt","yolo","yolov5","yolov9"],"latest_commit_sha":null,"homepage":"https://github.com/WongKinYiu/yolov9","language":"C++","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"mit","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/spacewalk01.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":"LICENSE","code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null}},"created_at":"2024-02-23T05:14:44.000Z","updated_at":"2024-10-15T05:09:57.000Z","dependencies_parsed_at":"2024-03-14T04:33:02.938Z","dependency_job_id":"0ac1a2ed-c388-477e-bc5f-197df7672677","html_url":"https://github.com/spacewalk01/tensorrt-yolov9","commit_stats":null,"previous_names":["spacewalk01/tensorrt-yolov9"],"tags_count":0,"template":false,"template_full_name":null,"repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/spacewalk01%2Ftensorrt-yolov9","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/spacewalk01%2Ftensorrt-yolov9/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/spacewalk01%2Ftensorrt-yolov9/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/spacewalk01%2Ftensorrt-yolov9/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/spacewalk01","download_url":"https://codeload.github.com/spacewalk01/tensorrt-yolov9/tar.gz/refs/heads/main","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":244301380,"owners_count":20430932,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["cpp","object-detection","tensorrt","yolo","yolov5","yolov9"],"created_at":"2024-07-31T03:00:45.833Z","updated_at":"2025-03-18T20:31:07.772Z","avatar_url":"https://github.com/spacewalk01.png","language":"C++","funding_links":[],"categories":["C++"],"sub_categories":[],"readme":"\u003cdiv align=\"center\"\u003e\n\nTensorRT-YOLOv9\n===========================\n\n[![python](https://img.shields.io/badge/python-3.10.12-green)](https://www.python.org/downloads/release/python-31012/)\n[![cuda](https://img.shields.io/badge/cuda-11.6-green)](https://developer.nvidia.com/cuda-downloads)\n[![trt](https://img.shields.io/badge/TRT-8.6-green)](https://developer.nvidia.com/tensorrt)\n[![mit](https://img.shields.io/badge/license-MIT-blue)](https://github.com/spacewalk01/TensorRT-YOLOv9/tree/main?tab=MIT-1-ov-file#readme)\n\n\u003cdiv align=\"left\"\u003e\n \nThis repo hosts a C++ and python implementation of the [YOLOv9](https://github.com/WongKinYiu/yolov9) state of the art object detection model, leveraging the TensorRT API for efficient real-time inference.\n\u003cp align=\"center\" margin: 0 auto;\u003e\n  \u003cimg src=\"assets/traffic.gif\" width=\"360px\" /\u003e\n  \u003cimg src=\"assets/parkinglot.gif\" width=\"360px\" /\u003e \n\u003c/p\u003e\n\n## 🚀 Usage\n\n#### Python\n\n``` shell\ncd \u003cthis project path\u003e/python\npython yolov9_trt.py --engine yolov9-c.engine --data images --outdir output\n```\n\n#### C++\n\n``` shell\ncd \u003cthis project path\u003e/build/release\n\n# infer an image\nyolov9-tensorrt.exe yolov9-c.engine test.jpg\n# infer a folder(images)\nyolov9-tensorrt.exe yolov9-c.engine data\n# infer a video\nyolov9-tensorrt.exe yolov9-c.engine test.mp4 # the video path\n```\n\n## 🛠️ Build\n\n#### Python\n\nThe following command will install tensorrt for python:\n\n``` shell\ncd \u003ctensorrt installation path\u003e/python\npip install cuda-python\npip install tensorrt-8.6.0-cp310-none-win_amd64.whl\npip install opencv-python\n```\n\n#### C++\n\nRefer to our [docs/INSTALL.md](https://github.com/spacewalk01/tensorrt-yolov9/blob/main/docs/INSTALL.md) for detailed installation instructions.\n\n- Cuda preprocessing: [main branch](https://github.com/spacewalk01/TensorRT-YOLOv9/tree/main)\n- Cpu preprocessing: [cpu-preprocessing branch](https://github.com/spacewalk01/TensorRT-YOLOv9/tree/cpu_preprocessing)\n  \n## 🖥️ Requirement\n   - TensorRT\n   - CUDA, CudaNN\n   - C++ compiler with C++17 or higher support\n   - CMake 3.14 or higher\n   - OpenCV\n\n## 🌱 Examples\n\nParameters for inference:\n\n``` \nConfidence threshold : 0.2\nNMS threshold : 0.3\nModel : yolov9-e (with FP32)\n```\n\nInference result:\n\n\u003cp align=\"center\" margin: 0 auto;\u003e\n  \u003cimg src=\"assets/street_o.jpg\" /\u003e \n\u003c/p\u003e\nOriginal image: https://www.flickr.com/photos/nicolelee/19041780\n\n## 👏 Acknowledgement\n\nThis project is based on the following awesome projects:\n- [YOLOv9](https://github.com/WongKinYiu/yolov9) - YOLOv9: Learning What You Want to Learn Using Programmable Gradient Information.\n- [TensorRT](https://github.com/NVIDIA/TensorRT/tree/release/8.6/samples) - TensorRT samples and api documentation.\n- [TensorRTx](https://github.com/wang-xinyu/tensorrtx) - Implementation of popular deep learning networks with TensorRT network definition API.\n\n## 🔗 See also\n- [Yolov9-Bytetrack](https://github.com/spacewalk01/yolov9-bytetrack-tensorrt) - Integration of YOLOv9 with ByteTracker using the TensorRT API.\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fspacewalk01%2Ftensorrt-yolov9","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fspacewalk01%2Ftensorrt-yolov9","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fspacewalk01%2Ftensorrt-yolov9/lists"}