cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Internal change

Browse Source 
PiperOrigin-RevId: 521406957
This commit is contained in:
MediaPipe Team 2023-04-03 02:58:49 -07:00 committed by Copybara-Service
parent 696bedcaa1
commit b5bbed8ebb
2 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
mediapipe/framework/formats/motion/optical_flow_field.cc
View File

@ -66,12 +66,12 @@ cv::Mat MakeVisualizationHsv(const cv::Mat_<float>& angles,
cv::Mat hsv(angles.size(), CV_8UC3); cv::Mat hsv(angles.size(), CV_8UC3);
for (int r = 0; r < hsv.rows; ++r) { for (int r = 0; r < hsv.rows; ++r) {
for (int c = 0; c < hsv.cols; ++c) { for (int c = 0; c < hsv.cols; ++c) {
const uint8 hue = static_cast<uint8>(255.0f * angles(r, c) / 360.0f); const uint8_t hue = static_cast<uint8_t>(255.0f * angles(r, c) / 360.0f);
uint8 saturation = 255; uint8_t saturation = 255;
if (magnitudes(r, c) < max_mag) { if (magnitudes(r, c) < max_mag) {
saturation = static_cast<uint8>(255.0f * magnitudes(r, c) / max_mag); saturation = static_cast<uint8_t>(255.0f * magnitudes(r, c) / max_mag);
} }
const uint8 value = 255; const uint8_t value = 255;
hsv.at<cv::Vec3b>(r, c) = cv::Vec3b(hue, saturation, value); hsv.at<cv::Vec3b>(r, c) = cv::Vec3b(hue, saturation, value);
} }
@ -282,7 +282,7 @@ void OpticalFlowField::EstimateMotionConsistencyOcclusions(
Location OpticalFlowField::FindMotionInconsistentPixels( Location OpticalFlowField::FindMotionInconsistentPixels(
const OpticalFlowField& forward, const OpticalFlowField& backward, const OpticalFlowField& forward, const OpticalFlowField& backward,
double spatial_distance_threshold) { double spatial_distance_threshold) {
const uint8 kOccludedPixelValue = 1; const uint8_t kOccludedPixelValue = 1;
const double threshold_sq = const double threshold_sq =
spatial_distance_threshold * spatial_distance_threshold; spatial_distance_threshold * spatial_distance_threshold;
cv::Mat occluded = cv::Mat::zeros(forward.height(), forward.width(), CV_8UC1); cv::Mat occluded = cv::Mat::zeros(forward.height(), forward.width(), CV_8UC1);
@ -301,10 +301,10 @@ Location OpticalFlowField::FindMotionInconsistentPixels(
if (!in_bounds_in_next_frame || if (!in_bounds_in_next_frame ||
Point2_f(x - round_trip_x, y - round_trip_y).ToVector().Norm2() > Point2_f(x - round_trip_x, y - round_trip_y).ToVector().Norm2() >
threshold_sq) { threshold_sq) {
occluded.at<uint8>(y, x) = kOccludedPixelValue; occluded.at<uint8_t>(y, x) = kOccludedPixelValue;
} }
} }
} }
return CreateCvMaskLocation<uint8>(occluded); return CreateCvMaskLocation<uint8_t>(occluded);
} }
} // namespace mediapipe } // namespace mediapipe

8
mediapipe/framework/formats/motion/optical_flow_field_test.cc
View File

@ -300,15 +300,15 @@ TEST(OpticalFlowField, Occlusions) {
for (int y = 0; y < occlusion_mat->rows; ++y) { for (int y = 0; y < occlusion_mat->rows; ++y) {
// Bottom row and pixel at (x, y) = (1, 0) are occluded. // Bottom row and pixel at (x, y) = (1, 0) are occluded.
if (y == occlusion_mat->rows - 1 || (x == 1 && y == 0)) { if (y == occlusion_mat->rows - 1 || (x == 1 && y == 0)) {
EXPECT_GT(occlusion_mat->at<uint8>(y, x), 0); EXPECT_GT(occlusion_mat->at<uint8_t>(y, x), 0);
} else { } else {
EXPECT_EQ(0, occlusion_mat->at<uint8>(y, x)); EXPECT_EQ(0, occlusion_mat->at<uint8_t>(y, x));
} }
// Top row and pixel at (x, y) = (1, 2) are disoccluded. // Top row and pixel at (x, y) = (1, 2) are disoccluded.
if (y == 0 || (x == 1 && y == 2)) { if (y == 0 || (x == 1 && y == 2)) {
EXPECT_GT(disocclusion_mat->at<uint8>(y, x), 0); EXPECT_GT(disocclusion_mat->at<uint8_t>(y, x), 0);
} else { } else {
EXPECT_EQ(0, disocclusion_mat->at<uint8>(y, x)); EXPECT_EQ(0, disocclusion_mat->at<uint8_t>(y, x));
} }
} }
} }