├── LICENSE ├── README.md ├── data ├── bus.jpg └── zidane.jpg ├── docker ├── Dockerfile └── readme.md ├── docs ├── API-Build.md ├── Jetson.md ├── Normal.md ├── Pose.md ├── Segment.md └── star.md ├── requirements.txt ├── scripts ├── .pre-commit-config.yaml ├── build.py ├── config.py ├── export-det.py ├── export-pose.py ├── export-seg.py ├── gen_pkl.py ├── infer-det-without-torch.py ├── infer-det.py ├── infer-pose-without-torch.py ├── infer-pose.py ├── infer-seg-without-torch.py ├── infer-seg.py ├── models │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-38.pyc │ │ ├── common.cpython-38.pyc │ │ └── engine.cpython-38.pyc │ ├── api.py │ ├── common.py │ ├── cudart_api.py │ ├── engine.py │ ├── pycuda_api.py │ ├── torch_utils.py │ └── utils.py └── trt-profile.py ├── src ├── deepstream │ ├── CMakeLists.txt │ ├── README.md │ ├── config_yoloV8.txt │ ├── custom_bbox_parser │ │ └── nvdsparsebbox_yoloV8.cpp │ ├── deepstream_app_config.txt │ └── labels.txt ├── detect │ ├── end2end │ │ ├── CMakeLists.txt │ │ ├── include │ │ │ ├── common.hpp │ │ │ └── yolov8.hpp │ │ └── main.cpp │ └── normal │ │ ├── CMakeLists.txt │ │ ├── include │ │ ├── common.hpp │ │ └── yolov8.hpp │ │ └── main.cpp ├── jetson │ ├── detect │ │ ├── CMakeLists.txt │ │ ├── include │ │ │ ├── common.hpp │ │ │ └── yolov8.hpp │ │ └── main.cpp │ ├── pose │ │ ├── CMakeLists.txt │ │ ├── include │ │ │ ├── common.hpp │ │ │ └── yolov8-pose.hpp │ │ └── main.cpp │ └── segment │ │ ├── CMakeLists.txt │ │ ├── include │ │ ├── common.hpp │ │ └── yolov8-seg.hpp │ │ └── main.cpp ├── pose │ └── normal │ │ ├── CMakeLists.txt │ │ ├── include │ │ ├── common.hpp │ │ └── yolov8-pose.hpp │ │ └── main.cpp └── segment │ ├── normal │ ├── CMakeLists.txt │ ├── include │ │ ├── common.hpp │ │ └── yolov8-seg.hpp │ └── main.cpp │ └── simple │ ├── CMakeLists.txt │ ├── include │ ├── common.hpp │ └── yolov8-seg.hpp │ └── main.cpp └── weights ├── .readme.md ├── yolov8n-pose.onnx ├── yolov8n-seg.onnx └── yolov8n.onnx /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/README.md -------------------------------------------------------------------------------- /data/bus.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/data/bus.jpg -------------------------------------------------------------------------------- /data/zidane.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/data/zidane.jpg -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /docker/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/docker/readme.md -------------------------------------------------------------------------------- /docs/API-Build.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/docs/API-Build.md -------------------------------------------------------------------------------- /docs/Jetson.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/docs/Jetson.md -------------------------------------------------------------------------------- /docs/Normal.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/docs/Normal.md -------------------------------------------------------------------------------- /docs/Pose.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/docs/Pose.md -------------------------------------------------------------------------------- /docs/Segment.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/docs/Segment.md -------------------------------------------------------------------------------- /docs/star.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/docs/star.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/.pre-commit-config.yaml -------------------------------------------------------------------------------- /scripts/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/build.py -------------------------------------------------------------------------------- /scripts/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/config.py -------------------------------------------------------------------------------- /scripts/export-det.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/export-det.py -------------------------------------------------------------------------------- /scripts/export-pose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/export-pose.py -------------------------------------------------------------------------------- /scripts/export-seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/export-seg.py -------------------------------------------------------------------------------- /scripts/gen_pkl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/gen_pkl.py -------------------------------------------------------------------------------- /scripts/infer-det-without-torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/infer-det-without-torch.py -------------------------------------------------------------------------------- /scripts/infer-det.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/infer-det.py -------------------------------------------------------------------------------- /scripts/infer-pose-without-torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/infer-pose-without-torch.py -------------------------------------------------------------------------------- /scripts/infer-pose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/infer-pose.py -------------------------------------------------------------------------------- /scripts/infer-seg-without-torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/infer-seg-without-torch.py -------------------------------------------------------------------------------- /scripts/infer-seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/infer-seg.py -------------------------------------------------------------------------------- /scripts/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/models/__init__.py -------------------------------------------------------------------------------- /scripts/models/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/models/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /scripts/models/__pycache__/common.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/models/__pycache__/common.cpython-38.pyc -------------------------------------------------------------------------------- /scripts/models/__pycache__/engine.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/models/__pycache__/engine.cpython-38.pyc -------------------------------------------------------------------------------- /scripts/models/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/models/api.py -------------------------------------------------------------------------------- /scripts/models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/models/common.py -------------------------------------------------------------------------------- /scripts/models/cudart_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/models/cudart_api.py -------------------------------------------------------------------------------- /scripts/models/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/models/engine.py -------------------------------------------------------------------------------- /scripts/models/pycuda_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/models/pycuda_api.py -------------------------------------------------------------------------------- /scripts/models/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/models/torch_utils.py -------------------------------------------------------------------------------- /scripts/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/models/utils.py -------------------------------------------------------------------------------- /scripts/trt-profile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/scripts/trt-profile.py -------------------------------------------------------------------------------- /src/deepstream/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/deepstream/CMakeLists.txt -------------------------------------------------------------------------------- /src/deepstream/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/deepstream/README.md -------------------------------------------------------------------------------- /src/deepstream/config_yoloV8.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/deepstream/config_yoloV8.txt -------------------------------------------------------------------------------- /src/deepstream/custom_bbox_parser/nvdsparsebbox_yoloV8.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/deepstream/custom_bbox_parser/nvdsparsebbox_yoloV8.cpp -------------------------------------------------------------------------------- /src/deepstream/deepstream_app_config.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/deepstream/deepstream_app_config.txt -------------------------------------------------------------------------------- /src/deepstream/labels.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/deepstream/labels.txt -------------------------------------------------------------------------------- /src/detect/end2end/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/detect/end2end/CMakeLists.txt -------------------------------------------------------------------------------- /src/detect/end2end/include/common.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/detect/end2end/include/common.hpp -------------------------------------------------------------------------------- /src/detect/end2end/include/yolov8.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/detect/end2end/include/yolov8.hpp -------------------------------------------------------------------------------- /src/detect/end2end/main.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/detect/end2end/main.cpp -------------------------------------------------------------------------------- /src/detect/normal/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/detect/normal/CMakeLists.txt -------------------------------------------------------------------------------- /src/detect/normal/include/common.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/detect/normal/include/common.hpp -------------------------------------------------------------------------------- /src/detect/normal/include/yolov8.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/detect/normal/include/yolov8.hpp -------------------------------------------------------------------------------- /src/detect/normal/main.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/detect/normal/main.cpp -------------------------------------------------------------------------------- /src/jetson/detect/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/jetson/detect/CMakeLists.txt -------------------------------------------------------------------------------- /src/jetson/detect/include/common.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/jetson/detect/include/common.hpp -------------------------------------------------------------------------------- /src/jetson/detect/include/yolov8.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/jetson/detect/include/yolov8.hpp -------------------------------------------------------------------------------- /src/jetson/detect/main.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/jetson/detect/main.cpp -------------------------------------------------------------------------------- /src/jetson/pose/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/jetson/pose/CMakeLists.txt -------------------------------------------------------------------------------- /src/jetson/pose/include/common.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/jetson/pose/include/common.hpp -------------------------------------------------------------------------------- /src/jetson/pose/include/yolov8-pose.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/jetson/pose/include/yolov8-pose.hpp -------------------------------------------------------------------------------- /src/jetson/pose/main.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/jetson/pose/main.cpp -------------------------------------------------------------------------------- /src/jetson/segment/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/jetson/segment/CMakeLists.txt -------------------------------------------------------------------------------- /src/jetson/segment/include/common.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/jetson/segment/include/common.hpp -------------------------------------------------------------------------------- /src/jetson/segment/include/yolov8-seg.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/jetson/segment/include/yolov8-seg.hpp -------------------------------------------------------------------------------- /src/jetson/segment/main.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/jetson/segment/main.cpp -------------------------------------------------------------------------------- /src/pose/normal/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/pose/normal/CMakeLists.txt -------------------------------------------------------------------------------- /src/pose/normal/include/common.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/pose/normal/include/common.hpp -------------------------------------------------------------------------------- /src/pose/normal/include/yolov8-pose.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/pose/normal/include/yolov8-pose.hpp -------------------------------------------------------------------------------- /src/pose/normal/main.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/pose/normal/main.cpp -------------------------------------------------------------------------------- /src/segment/normal/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/segment/normal/CMakeLists.txt -------------------------------------------------------------------------------- /src/segment/normal/include/common.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/segment/normal/include/common.hpp -------------------------------------------------------------------------------- /src/segment/normal/include/yolov8-seg.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/segment/normal/include/yolov8-seg.hpp -------------------------------------------------------------------------------- /src/segment/normal/main.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/segment/normal/main.cpp -------------------------------------------------------------------------------- /src/segment/simple/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/segment/simple/CMakeLists.txt -------------------------------------------------------------------------------- /src/segment/simple/include/common.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/segment/simple/include/common.hpp -------------------------------------------------------------------------------- /src/segment/simple/include/yolov8-seg.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/segment/simple/include/yolov8-seg.hpp -------------------------------------------------------------------------------- /src/segment/simple/main.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/src/segment/simple/main.cpp -------------------------------------------------------------------------------- /weights/.readme.md: -------------------------------------------------------------------------------- 1 | + 权重下载后放此目录下 -------------------------------------------------------------------------------- /weights/yolov8n-pose.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/weights/yolov8n-pose.onnx -------------------------------------------------------------------------------- /weights/yolov8n-seg.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/weights/yolov8n-seg.onnx -------------------------------------------------------------------------------- /weights/yolov8n.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linClubs/YOLOv8-ROS-TensorRT/HEAD/weights/yolov8n.onnx --------------------------------------------------------------------------------