Skip to content

Commit

Permalink
fix readme
Browse files Browse the repository at this point in the history
  • Loading branch information
Ar-Ray-code committed Apr 26, 2022
1 parent 4d1b76d commit 6763934
Showing 1 changed file with 16 additions and 4 deletions.
20 changes: 16 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -58,9 +58,15 @@ pip3 install yolox
```bash
source /opt/ros/foxy/setup.bash
sudo apt install ros-foxy-v4l2-camera
# source /opt/intel/openvino_2021/bin/setupvars.sh # <- Using OpenVINO
colcon build --symlink-install # weights (YOLOX-Nano) files will be installed automatically.
```

**Automatic download weights**

- yolox_nano.onnx
- yolox_nano.pth

### (Step 3) Using CUDA

If you have NVIDIA Graphics, you can run YOLOX-ROS on GPU.
Expand All @@ -71,20 +77,26 @@ If you have NVIDIA Graphics, you can run YOLOX-ROS on GPU.
- CUDA toolkit (11.0)
- torch+cuda

```bash
source /opt/ros/foxy/setup.bash
sudo apt install ros-foxy-v4l2-camera
colcon build --symlink-install # weights (YOLOX-Nano) files will be installed automatically.
```

### Step : Demo

Connect your web camera.

```bash
source /opt/ros/foxy/setup.bash
source ~/ros2_ws/install/local_setup.bash
ros2 launch yolox_ros_py yolox_nano_cpu.launch.py
# ros2 launch yolox_ros_py yolox_nano.launch.py # <- GPU
ros2 launch yolox_ros_py yolox_nano_cpu.launch.py # <- CPU (PyTorch)
# ros2 launch yolox_ros_py yolox_nano.launch.py # <- GPU (PyTorch)
# ros2 launch yolox_ros_py yolox_nano_onnx.launch.py # <- ONNXRuntime

# OpenVINO ---
# OpenVINO -------------------------------------
# source /opt/intel/openvino_2021/bin/setupvars.sh
# ros2 launch yolox_ros_py yolox_nano_openvino.launch.py # <- OpenVINO
# ros2 launch yolox_ros_py yolox_nano_openvino.launch.py
```

</details>
Expand Down

0 comments on commit 6763934

Please sign in to comment.