From e4693eb5fa765bc2e5c0d8d56577b1cdda695c43 Mon Sep 17 00:00:00 2001 From: Marko Bjelonic Date: Thu, 8 Apr 2021 14:15:16 +0200 Subject: [PATCH] Updated readme. --- README.md | 2 +- darknet_ros/CMakeLists.txt | 5 +++++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 68baa0cc1..b687b5170 100644 --- a/README.md +++ b/README.md @@ -4,7 +4,7 @@ This is a ROS package developed for object detection in camera images. You only look once (YOLO) is a state-of-the-art, real-time object detection system. In the following ROS package you are able to use YOLO (V3) on GPU and CPU. The pre-trained model of the convolutional neural network is able to detect pre-trained classes including the data set from VOC and COCO, or you can also create a network with your own detection objects. For more information about YOLO, Darknet, available training data and training YOLO see the following link: [YOLO: Real-Time Object Detection](http://pjreddie.com/darknet/yolo/). -The YOLO packages have been tested under ROS Melodic and Ubuntu 18.04. This is research code, expect that it changes often and any fitness for a particular purpose is disclaimed. +The YOLO packages have been tested under ROS Noetic and Ubuntu 20.04 (Please check the branches for melodic, foxy, and ROS2 versions). This is research code, expect that it changes often and any fitness for a particular purpose is disclaimed. **Author: [Marko Bjelonic](https://www.markobjelonic.com), marko.bjelonic@mavt.ethz.ch** diff --git a/darknet_ros/CMakeLists.txt b/darknet_ros/CMakeLists.txt index 807b0e571..919eb3b87 100644 --- a/darknet_ros/CMakeLists.txt +++ b/darknet_ros/CMakeLists.txt @@ -23,7 +23,12 @@ if (CUDA_FOUND) CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS}; -O3 + -gencode arch=compute_30,code=sm_30 + -gencode arch=compute_35,code=sm_35 + -gencode arch=compute_50,code=[sm_50,compute_50] -gencode arch=compute_52,code=[sm_52,compute_52] + -gencode arch=compute_61,code=sm_61 + -gencode arch=compute_62,code=sm_62 ) add_definitions(-DGPU) else()