Fixed color encoding for cv::imshow.

begger123 · Nov 20, 2017 · 87f86f7 · 87f86f7
1 parent 9f02598
commit 87f86f7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/darknet_ros/src/YoloObjectDetector.cpp b/darknet_ros/src/YoloObjectDetector.cpp
@@ -288,7 +288,9 @@ void YoloObjectDetector::runYolo(cv::Mat &fullFrame, const std_msgs::Header& hea
   }
 
   if(viewImage_ && !darknetImageViewer_) {
-    cv::imshow(opencvWindow_, inputFrame);
+    cv::Mat showImage;
+    cv::cvtColor(inputFrame, showImage, cv::COLOR_RGB2BGR);
+    cv::imshow(opencvWindow_, showImage);
     cv::waitKey(waitKeyDelay_);
   }
 

diff --git a/darknet_ros/test/ObjectDetection.cpp b/darknet_ros/test/ObjectDetection.cpp
@@ -82,7 +82,7 @@ bool sendImageToYolo(ros::NodeHandle nh, std::string imageName) {
   // Get test image
   cv_bridge::CvImagePtr cv_ptr(new cv_bridge::CvImage);
   cv_ptr->image = cv::imread(pathToTestImage, CV_LOAD_IMAGE_COLOR);
-  cv_ptr->encoding = sensor_msgs::image_encodings::BGR8;
+  cv_ptr->encoding = sensor_msgs::image_encodings::RGB8;
   sensor_msgs::ImagePtr image = cv_ptr->toImageMsg();
 
   // Generate goal.