@@ -79,7 +79,7 @@ class SimpleFlowNodelet : public opencv_apps::Nodelet
7979 static bool need_config_update_;
8080 int scale_;
8181
82- cv::Mat gray, prevGray ;
82+ cv::Mat color, prevColor ;
8383
8484 void reconfigureCallback (Config& new_config, uint32_t level)
8585 {
@@ -117,7 +117,7 @@ class SimpleFlowNodelet : public opencv_apps::Nodelet
117117 // Convert the image into something opencv can handle.
118118 cv::Mat frame_src = cv_bridge::toCvShare (msg, sensor_msgs::image_encodings::BGR8)->image ;
119119
120- // / Convert it to gray
120+ // / Convert it to 3 channel
121121 cv::Mat frame;
122122 if (frame_src.channels () > 1 )
123123 {
@@ -128,18 +128,18 @@ class SimpleFlowNodelet : public opencv_apps::Nodelet
128128 cv::cvtColor (frame_src, frame, cv::COLOR_GRAY2BGR);
129129 }
130130
131- cv::resize (frame, gray , cv::Size (frame.cols / (double )MAX (1 , scale_), frame.rows / (double )MAX (1 , scale_)), 0 , 0 ,
131+ cv::resize (frame, color , cv::Size (frame.cols / (double )MAX (1 , scale_), frame.rows / (double )MAX (1 , scale_)), 0 , 0 ,
132132 CV_INTER_AREA);
133- if (prevGray .empty ())
134- gray .copyTo (prevGray );
133+ if (prevColor .empty ())
134+ color .copyTo (prevColor );
135135
136- if (gray .rows != prevGray .rows && gray .cols != prevGray .cols )
136+ if (color .rows != prevColor .rows && color .cols != prevColor .cols )
137137 {
138138 NODELET_WARN (" Images should be of equal sizes" );
139- gray .copyTo (prevGray );
139+ color .copyTo (prevColor );
140140 }
141141
142- if (frame.type () != 16 )
142+ if (frame.type () != CV_8UC3 )
143143 {
144144 NODELET_ERROR (" Images should be of equal type CV_8UC3" );
145145 }
@@ -164,9 +164,9 @@ class SimpleFlowNodelet : public opencv_apps::Nodelet
164164
165165 float start = (float )cv::getTickCount ();
166166#if CV_MAJOR_VERSION == 3
167- cv::optflow::calcOpticalFlowSF (gray, prevGray ,
167+ cv::optflow::calcOpticalFlowSF (color, prevColor ,
168168#else
169- cv::calcOpticalFlowSF (gray, prevGray ,
169+ cv::calcOpticalFlowSF (color, prevColor ,
170170#endif
171171 flow, 3 , 2 , 4 , 4.1 , 25.5 , 18 , 55.0 , 25.5 , 0.35 , 18 , 55.0 , 25.5 , 10 );
172172 NODELET_INFO (" calcOpticalFlowSF : %lf sec" , (cv::getTickCount () - start) / cv::getTickFrequency ());
@@ -210,7 +210,7 @@ class SimpleFlowNodelet : public opencv_apps::Nodelet
210210 int c = cv::waitKey (1 );
211211 }
212212
213- cv::swap (prevGray, gray );
213+ cv::swap (prevColor, color );
214214 // Publish the image.
215215 sensor_msgs::Image::Ptr out_img = cv_bridge::CvImage (msg->header , " bgr8" , frame).toImageMsg ();
216216 img_pub_.publish (out_img);
0 commit comments