|
7 | 7 | - TensorRT 8.x * |
8 | 8 | - ONNXRuntime * |
9 | 9 | - Tensorflow Lite * |
| 10 | +- **CUDA 11** |
10 | 11 |
|
11 | 12 | ※ Either one of OpenVINO or TensorRT or ONNXRuntime or Tensorflow Lite is required. |
12 | 13 |
|
|
18 | 19 |
|
19 | 20 | ※ Model convert script is not supported OpenVINO 2022.* |
20 | 21 |
|
21 | | -※ YOLOX is not required. |
| 22 | +※ Don't use CUDA 12 |
22 | 23 |
|
23 | | -※ Jetson + TensorRT docker support (Jetpack 4.6 r32.6.1). Tested with Jetson Nano 4GB. |
24 | | - |
25 | | - |
26 | | - |
27 | | - |
28 | | -<!-- <details> |
29 | | -<summary>Execute with docker</summary> |
30 | | -
|
31 | | -
|
32 | | -#### OpenVINO |
33 | | -```bash |
34 | | -# base image is "openvino/ubuntu20_dev:2021.4.1_20210416" |
35 | | -docker pull fateshelled/openvino_yolox_ros:latest |
36 | | -
|
37 | | -xhost + |
38 | | -docker run --rm -it \ |
39 | | - --network host \ |
40 | | - --privileged \ |
41 | | - --user openvino \ |
42 | | - -v $HOME/ros2_ws:/home/openvino/ros2_ws \ |
43 | | - -v /tmp/.X11-unix:/tmp/.X11-unix \ |
44 | | - -w /home/openvino/ros2_ws \ |
45 | | - -e DISPLAY=$DISPLAY \ |
46 | | - --device /dev/video0:/dev/video0 \ |
47 | | - fateshelled/openvino_yolox_ros:latest /bin/bash |
48 | | -
|
49 | | -# If use NCS2, mount "/dev/bus/usb". |
50 | | -xhost + |
51 | | -docker run --rm -it \ |
52 | | - --network host \ |
53 | | - --privileged \ |
54 | | - --user openvino \ |
55 | | - -v $HOME/ros2_ws:/home/openvino/ros2_ws \ |
56 | | - -v /tmp/.X11-unix:/tmp/.X11-unix \ |
57 | | - -w /home/openvino/ros2_ws \ |
58 | | - -v /dev/bus/usb:/dev/bus/usb |
59 | | - -e DISPLAY=$DISPLAY \ |
60 | | - --device /dev/video0:/dev/video0 \ |
61 | | - fateshelled/openvino_yolox_ros:latest \ |
62 | | - /bin/bash |
63 | | -
|
64 | | -``` |
65 | | -
|
66 | | -#### TensorRT |
67 | | -```bash |
68 | | -# base image is "nvcr.io/nvidia/pytorch:21.09-py3" |
69 | | -docker pull swiftfile/tensorrt_yolox_ros:latest |
70 | | -
|
71 | | -xhost + |
72 | | -docker run --rm -it \ |
73 | | - --network host \ |
74 | | - --gpus all \ |
75 | | - --privileged \ |
76 | | - -v $HOME/ros2_ws:/root/ros2_ws \ |
77 | | - -v /tmp/.X11-unix:/tmp/.X11-unix \ |
78 | | - -w /root/ros2_ws \ |
79 | | - -e DISPLAY=$DISPLAY \ |
80 | | - --device /dev/video0:/dev/video0 \ |
81 | | - swiftfile/tensorrt_yolox_ros:latest \ |
82 | | - /bin/bash |
83 | | -``` |
84 | | -
|
85 | | -#### Jetson + TensorRT |
86 | | -```bash |
87 | | -# base image is "dustynv/ros:foxy-ros-base-l4t-r32.6.1" |
88 | | -docker pull fateshelled/jetson_yolox_ros:foxy-ros-base-l4t-r32.6.1 |
89 | | -
|
90 | | -# This image cannot display GUI. |
91 | | -docker run --rm -it \ |
92 | | - --network host \ |
93 | | - --runtime nvidia \ |
94 | | - -v $HOME/ros2_ws:/root/ros2_ws \ |
95 | | - -w /root/ros2_ws \ |
96 | | - --device /dev/video0:/dev/video0 \ |
97 | | - fateshelled/jetson_yolox_ros:foxy-ros-base-l4t-r32.6.1 \ |
98 | | - /bin/bash |
99 | | -``` |
100 | | -
|
101 | | -#### ONNXRuntime |
102 | | -```bash |
103 | | -# base image is "nvcr.io/nvidia/cuda:11.4.2-cudnn8-devel-ubuntu20.04" |
104 | | -docker pull fateshelled/onnxruntime_yolox_ros:latest |
105 | | -
|
106 | | -xhost + |
107 | | -docker run --rm -it \ |
108 | | - --network host \ |
109 | | - --gpus all \ |
110 | | - --privileged \ |
111 | | - -v $HOME/ros2_ws:/root/ros2_ws \ |
112 | | - -v /tmp/.X11-unix:/tmp/.X11-unix \ |
113 | | - -w /root/ros2_ws \ |
114 | | - -e DISPLAY=$DISPLAY \ |
115 | | - --device /dev/video0:/dev/video0 \ |
116 | | - fateshelled/onnxruntime_yolox_ros:latest \ |
117 | | - /bin/bash |
118 | | -``` |
119 | | -
|
120 | | -</details> --> |
121 | 24 |
|
122 | 25 | ## Clone YOLOX-ROS |
123 | 26 | ```bash |
|
0 commit comments