YOLO-ROS for PyTorch

YOLO-ROS for TensorRT, please checkout for branch tensorrt

YOLO-ROS for HUAWEI ATLAS200, please checkout for branch atlas200

Modified from ros-yolov5, originated from yolov5

1. clone YOLO-ROS

   cd {YOUR_WORKSPACE}/src
   git clone https://github.com/jianhengLiu/yolo_ros.git
   

2. install python dependencies

   sudo apt install ros-melodic-ros-numpy
   pip3 install --upgrade pip
   # conda create -n dvins python=3.6
   # conda activate dvins
   pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu113
   pip3 install -r yolo_ros/requirements.txt

3. catkin_make

4. roslaunch yolo_ros yolo_service.launch

5. play your rosbag.