├── .gitignore ├── CMakeLists.txt ├── LICENSE ├── README.md ├── dependencies ├── install_protobuf-3.8.0.sh └── install_pycuda.sh ├── docs ├── results.gif ├── yolov4_custom_1.png ├── yolov4_custom_2.png └── yolov4_custom_3.png ├── launch ├── video_source.launch ├── yolov3_trt.launch ├── yolov4_trt.launch └── yolov4_trt_batch.launch ├── msg ├── BoundingBox2D.msg ├── Detector2D.msg ├── Detector2DArray.msg └── ObjectHypothesis.msg ├── package.xml ├── plugins ├── Makefile ├── README.md ├── libyolo_layer.so ├── yolo_layer.cu ├── yolo_layer.h └── yolo_layer.o ├── setup.py ├── src ├── image_converter.cpp ├── image_converter.h ├── node_video_source.cpp ├── ros_compat.cpp └── ros_compat.h ├── trt_yolo_v3.py ├── trt_yolo_v4.py ├── trt_yolo_v4_batch.py ├── utils ├── __init__.py ├── display.py ├── visualization.py ├── yolo_classes.py ├── yolo_with_plugins.py └── yolo_with_plugins_batch.py └── yolo ├── __pycache__ └── plugins.cpython-36.pyc ├── convert_yolo_trt.sh ├── download_yolo.sh ├── onnx_to_tensorrt.py ├── plugins.py ├── requirements.txt └── yolo_to_onnx.py /.gitignore: -------------------------------------------------------------------------------- 1 | yolo/yolov4-416.trt 2 | -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/README.md -------------------------------------------------------------------------------- /dependencies/install_protobuf-3.8.0.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/dependencies/install_protobuf-3.8.0.sh -------------------------------------------------------------------------------- /dependencies/install_pycuda.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/dependencies/install_pycuda.sh -------------------------------------------------------------------------------- /docs/results.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/docs/results.gif -------------------------------------------------------------------------------- /docs/yolov4_custom_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/docs/yolov4_custom_1.png -------------------------------------------------------------------------------- /docs/yolov4_custom_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/docs/yolov4_custom_2.png -------------------------------------------------------------------------------- /docs/yolov4_custom_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/docs/yolov4_custom_3.png -------------------------------------------------------------------------------- /launch/video_source.launch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/launch/video_source.launch -------------------------------------------------------------------------------- /launch/yolov3_trt.launch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/launch/yolov3_trt.launch -------------------------------------------------------------------------------- /launch/yolov4_trt.launch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/launch/yolov4_trt.launch -------------------------------------------------------------------------------- /launch/yolov4_trt_batch.launch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/launch/yolov4_trt_batch.launch -------------------------------------------------------------------------------- /msg/BoundingBox2D.msg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/msg/BoundingBox2D.msg -------------------------------------------------------------------------------- /msg/Detector2D.msg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/msg/Detector2D.msg -------------------------------------------------------------------------------- /msg/Detector2DArray.msg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/msg/Detector2DArray.msg -------------------------------------------------------------------------------- /msg/ObjectHypothesis.msg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/msg/ObjectHypothesis.msg -------------------------------------------------------------------------------- /package.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/package.xml -------------------------------------------------------------------------------- /plugins/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/plugins/Makefile -------------------------------------------------------------------------------- /plugins/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/plugins/README.md -------------------------------------------------------------------------------- /plugins/libyolo_layer.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/plugins/libyolo_layer.so -------------------------------------------------------------------------------- /plugins/yolo_layer.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/plugins/yolo_layer.cu -------------------------------------------------------------------------------- /plugins/yolo_layer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/plugins/yolo_layer.h -------------------------------------------------------------------------------- /plugins/yolo_layer.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/plugins/yolo_layer.o -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/setup.py -------------------------------------------------------------------------------- /src/image_converter.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/src/image_converter.cpp -------------------------------------------------------------------------------- /src/image_converter.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/src/image_converter.h -------------------------------------------------------------------------------- /src/node_video_source.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/src/node_video_source.cpp -------------------------------------------------------------------------------- /src/ros_compat.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/src/ros_compat.cpp -------------------------------------------------------------------------------- /src/ros_compat.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/src/ros_compat.h -------------------------------------------------------------------------------- /trt_yolo_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/trt_yolo_v3.py -------------------------------------------------------------------------------- /trt_yolo_v4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/trt_yolo_v4.py -------------------------------------------------------------------------------- /trt_yolo_v4_batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/trt_yolo_v4_batch.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/display.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/utils/display.py -------------------------------------------------------------------------------- /utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/utils/visualization.py -------------------------------------------------------------------------------- /utils/yolo_classes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/utils/yolo_classes.py -------------------------------------------------------------------------------- /utils/yolo_with_plugins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/utils/yolo_with_plugins.py -------------------------------------------------------------------------------- /utils/yolo_with_plugins_batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/utils/yolo_with_plugins_batch.py -------------------------------------------------------------------------------- /yolo/__pycache__/plugins.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/yolo/__pycache__/plugins.cpython-36.pyc -------------------------------------------------------------------------------- /yolo/convert_yolo_trt.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/yolo/convert_yolo_trt.sh -------------------------------------------------------------------------------- /yolo/download_yolo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/yolo/download_yolo.sh -------------------------------------------------------------------------------- /yolo/onnx_to_tensorrt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/yolo/onnx_to_tensorrt.py -------------------------------------------------------------------------------- /yolo/plugins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/yolo/plugins.py -------------------------------------------------------------------------------- /yolo/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/yolo/requirements.txt -------------------------------------------------------------------------------- /yolo/yolo_to_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/indra4837/yolov4_trt_ros/HEAD/yolo/yolo_to_onnx.py --------------------------------------------------------------------------------