30 lines (22 loc) · 829 Bytes

YOLO-ROS

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

clone YOLO-ROS

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

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

catkin_make
roslaunch yolo_ros yolo_service.launch
play your rosbag.