experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fixed the sanity test problem

Browse Source
This commit is contained in:
Koan-Sin Tan 2020-03-11 15:07:16 +08:00
parent 06453ac4db
commit a1c525f01c
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/lite/examples/label_image/label_image_test.cc
View File

@ -29,6 +29,7 @@ TEST(LabelImageTest, GraceHopper) {
"grace_hopper.bmp"; "grace_hopper.bmp";
int height, width, channels; int height, width, channels;
Settings s; Settings s;
s.input_type = kTfLiteUInt8;
std::vector<uint8_t> input = std::vector<uint8_t> input =
read_bmp(lena_file, &width, &height, &channels, &s); read_bmp(lena_file, &width, &height, &channels, &s);
ASSERT_EQ(height, 606); ASSERT_EQ(height, 606);
@ -37,8 +38,8 @@ TEST(LabelImageTest, GraceHopper) {
std::vector<uint8_t> output(606 * 517 * 3); std::vector<uint8_t> output(606 * 517 * 3);
resize<uint8_t>(output.data(), input.data(), 606, 517, 3, 214, 214, 3, &s); resize<uint8_t>(output.data(), input.data(), 606, 517, 3, 214, 214, 3, &s);
ASSERT_EQ(output[0], 0x0); ASSERT_EQ(output[0], 0x15);
ASSERT_EQ(output[214 * 214 * 3 - 1], 0x0); ASSERT_EQ(output[214 * 214 * 3 - 1], 0x11);
} }
TEST(LabelImageTest, GetTopN) { TEST(LabelImageTest, GetTopN) {