├── Dockerfile ├── LICENSE ├── README.md ├── bus.jpg ├── cpp ├── CMakeLists.txt ├── libtorch │ ├── yolo_libtorch.cpp │ └── yolo_libtorch.h ├── magic_enum.hpp ├── main.cpp ├── onnxruntime │ ├── yolo_onnxruntime.cpp │ └── yolo_onnxruntime.h ├── opencv │ ├── yolo_opencv.cpp │ └── yolo_opencv.h ├── openvino │ ├── yolo_openvino.cpp │ └── yolo_openvino.h ├── run.bat ├── run.sh ├── tensorrt │ ├── decode.cu │ ├── decode.cuh │ ├── preprocess.cu │ ├── preprocess.cuh │ ├── yolo_tensorrt.cpp │ └── yolo_tensorrt.h ├── utils.cpp ├── utils.h ├── yolo.cpp ├── yolo.h ├── yolo_classify.h ├── yolo_detect.h └── yolo_segment.h └── python ├── backends ├── ONNXRuntime │ ├── __init__.py │ └── yolo_onnxruntime.py ├── OpenCV │ ├── __init__.py │ └── yolo_opencv.py ├── OpenVINO │ ├── __init__.py │ └── yolo_openvino.py ├── PyTorch │ ├── __init__.py │ └── yolo_pytorch.py ├── TensorRT │ ├── __init__.py │ └── yolo_tensorrt.py ├── __init__.py ├── utils.py └── yolo.py ├── main.py ├── requirements.txt ├── run.bat └── run.sh /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/README.md -------------------------------------------------------------------------------- /bus.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/bus.jpg -------------------------------------------------------------------------------- /cpp/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/CMakeLists.txt -------------------------------------------------------------------------------- /cpp/libtorch/yolo_libtorch.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/libtorch/yolo_libtorch.cpp -------------------------------------------------------------------------------- /cpp/libtorch/yolo_libtorch.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/libtorch/yolo_libtorch.h -------------------------------------------------------------------------------- /cpp/magic_enum.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/magic_enum.hpp -------------------------------------------------------------------------------- /cpp/main.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/main.cpp -------------------------------------------------------------------------------- /cpp/onnxruntime/yolo_onnxruntime.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/onnxruntime/yolo_onnxruntime.cpp -------------------------------------------------------------------------------- /cpp/onnxruntime/yolo_onnxruntime.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/onnxruntime/yolo_onnxruntime.h -------------------------------------------------------------------------------- /cpp/opencv/yolo_opencv.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/opencv/yolo_opencv.cpp -------------------------------------------------------------------------------- /cpp/opencv/yolo_opencv.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/opencv/yolo_opencv.h -------------------------------------------------------------------------------- /cpp/openvino/yolo_openvino.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/openvino/yolo_openvino.cpp -------------------------------------------------------------------------------- /cpp/openvino/yolo_openvino.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/openvino/yolo_openvino.h -------------------------------------------------------------------------------- /cpp/run.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/run.bat -------------------------------------------------------------------------------- /cpp/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/run.sh -------------------------------------------------------------------------------- /cpp/tensorrt/decode.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/tensorrt/decode.cu -------------------------------------------------------------------------------- /cpp/tensorrt/decode.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/tensorrt/decode.cuh -------------------------------------------------------------------------------- /cpp/tensorrt/preprocess.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/tensorrt/preprocess.cu -------------------------------------------------------------------------------- /cpp/tensorrt/preprocess.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/tensorrt/preprocess.cuh -------------------------------------------------------------------------------- /cpp/tensorrt/yolo_tensorrt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/tensorrt/yolo_tensorrt.cpp -------------------------------------------------------------------------------- /cpp/tensorrt/yolo_tensorrt.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/tensorrt/yolo_tensorrt.h -------------------------------------------------------------------------------- /cpp/utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/utils.cpp -------------------------------------------------------------------------------- /cpp/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/utils.h -------------------------------------------------------------------------------- /cpp/yolo.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/yolo.cpp -------------------------------------------------------------------------------- /cpp/yolo.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/yolo.h -------------------------------------------------------------------------------- /cpp/yolo_classify.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/yolo_classify.h -------------------------------------------------------------------------------- /cpp/yolo_detect.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/yolo_detect.h -------------------------------------------------------------------------------- /cpp/yolo_segment.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/cpp/yolo_segment.h -------------------------------------------------------------------------------- /python/backends/ONNXRuntime/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/python/backends/ONNXRuntime/__init__.py -------------------------------------------------------------------------------- /python/backends/ONNXRuntime/yolo_onnxruntime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/python/backends/ONNXRuntime/yolo_onnxruntime.py -------------------------------------------------------------------------------- /python/backends/OpenCV/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/python/backends/OpenCV/__init__.py -------------------------------------------------------------------------------- /python/backends/OpenCV/yolo_opencv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/python/backends/OpenCV/yolo_opencv.py -------------------------------------------------------------------------------- /python/backends/OpenVINO/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/python/backends/OpenVINO/__init__.py -------------------------------------------------------------------------------- /python/backends/OpenVINO/yolo_openvino.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/python/backends/OpenVINO/yolo_openvino.py -------------------------------------------------------------------------------- /python/backends/PyTorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/python/backends/PyTorch/__init__.py -------------------------------------------------------------------------------- /python/backends/PyTorch/yolo_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/python/backends/PyTorch/yolo_pytorch.py -------------------------------------------------------------------------------- /python/backends/TensorRT/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/python/backends/TensorRT/__init__.py -------------------------------------------------------------------------------- /python/backends/TensorRT/yolo_tensorrt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/python/backends/TensorRT/yolo_tensorrt.py -------------------------------------------------------------------------------- /python/backends/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/python/backends/__init__.py -------------------------------------------------------------------------------- /python/backends/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/python/backends/utils.py -------------------------------------------------------------------------------- /python/backends/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/python/backends/yolo.py -------------------------------------------------------------------------------- /python/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/python/main.py -------------------------------------------------------------------------------- /python/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/python/requirements.txt -------------------------------------------------------------------------------- /python/run.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/python/run.bat -------------------------------------------------------------------------------- /python/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taifyang/yolo-inference/HEAD/python/run.sh --------------------------------------------------------------------------------