talksik/mediapipe
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Internal change

Browse Source 
PiperOrigin-RevId: 525390694
This commit is contained in:
MediaPipe Team
2023-04-19 02:39:54 -07:00
committed by Copybara-Service
parent b83fa5b67d
commit de84696be6
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/tasks/cc/components/processors/classification_postprocessing_graph_test.cc
+5 -5
View File
@@ -520,7 +520,7 @@ TEST_F(PostprocessingTest, SucceedsWithoutMetadata) {
auto poller, auto poller,
BuildGraph(kQuantizedImageClassifierWithoutMetadata, options)); BuildGraph(kQuantizedImageClassifierWithoutMetadata, options));
// Build input tensors. // Build input tensors.
std::vector<uint8> tensor(kMobileNetNumClasses, 0); std::vector<uint8_t> tensor(kMobileNetNumClasses, 0);
tensor[1] = 18; tensor[1] = 18;
tensor[2] = 16; tensor[2] = 16;
@@ -552,7 +552,7 @@ TEST_F(PostprocessingTest, SucceedsWithMetadata) {
MP_ASSERT_OK_AND_ASSIGN( MP_ASSERT_OK_AND_ASSIGN(
auto poller, BuildGraph(kQuantizedImageClassifierWithMetadata, options)); auto poller, BuildGraph(kQuantizedImageClassifierWithMetadata, options));
// Build input tensors. // Build input tensors.
std::vector<uint8> tensor(kMobileNetNumClasses, 0); std::vector<uint8_t> tensor(kMobileNetNumClasses, 0);
tensor[1] = 12; tensor[1] = 12;
tensor[2] = 14; tensor[2] = 14;
tensor[3] = 16; tensor[3] = 16;
@@ -589,7 +589,7 @@ TEST_F(PostprocessingTest, SucceedsWithScoreCalibration) {
auto poller, auto poller,
BuildGraph(kQuantizedImageClassifierWithDummyScoreCalibration, options)); BuildGraph(kQuantizedImageClassifierWithDummyScoreCalibration, options));
// Build input tensors. // Build input tensors.
std::vector<uint8> tensor(kMobileNetNumClasses, 0); std::vector<uint8_t> tensor(kMobileNetNumClasses, 0);
tensor[1] = 12; tensor[1] = 12;
tensor[2] = 14; tensor[2] = 14;
tensor[3] = 16; tensor[3] = 16;
@@ -677,11 +677,11 @@ TEST_F(PostprocessingTest, SucceedsWithTimestamps) {
auto poller, BuildGraph(kQuantizedImageClassifierWithMetadata, options, auto poller, BuildGraph(kQuantizedImageClassifierWithMetadata, options,
/*connect_timestamps=*/true)); /*connect_timestamps=*/true));
// Build input tensors. // Build input tensors.
std::vector<uint8> tensor_0(kMobileNetNumClasses, 0); std::vector<uint8_t> tensor_0(kMobileNetNumClasses, 0);
tensor_0[1] = 12; tensor_0[1] = 12;
tensor_0[2] = 14; tensor_0[2] = 14;
tensor_0[3] = 16; tensor_0[3] = 16;
std::vector<uint8> tensor_1(kMobileNetNumClasses, 0); std::vector<uint8_t> tensor_1(kMobileNetNumClasses, 0);
tensor_1[5] = 12; tensor_1[5] = 12;
tensor_1[6] = 14; tensor_1[6] = 14;
tensor_1[7] = 16; tensor_1[7] = 16;