Browse Source

fix(lite/load_and_run): fix suffix parse error

GitOrigin-RevId: 8a9dac219e
tags/v1.8.0
Megvii Engine Team 3 years ago
parent
commit
099d1b87f6
1 changed files with 6 additions and 1 deletions
  1. +6
    -1
      lite/load_and_run/src/helpers/data_parser.cpp

+ 6
- 1
lite/load_and_run/src/helpers/data_parser.cpp View File

@@ -30,7 +30,12 @@ void DataParser::feed(const std::string& path) {
}

auto endWith = [blob_string](std::string suffix) -> bool {
return blob_string.rfind(suffix) == (blob_string.length() - suffix.length());
const auto index = blob_string.rfind(suffix);
if (index != std::string::npos and
index == blob_string.length() - suffix.length()) {
return true;
}
return false;
};

if (endWith(".ppm") || endWith(".pgm")) {


Loading…
Cancel
Save