supplement path modeify for windows

4 years ago · 0e9b66a03b
parent c88ed51c99
commit 0e9b66a03b
1 changed files with 5 additions and 0 deletions
--- a/mindspore/lite/tools/converter/converter_flags.cc
+++ b/mindspore/lite/tools/converter/converter_flags.cc
@ -17,6 +17,7 @@
 #include "tools/converter/converter_flags.h"
 #include <regex>
 #include <string>
+#include <algorithm>
 #include "ir/dtype/type_id.h"

 namespace mindspore {
@ -78,6 +79,10 @@ int Flags::Init(int argc, const char **argv) {
    return RET_INPUT_PARAM_INVALID;
  }

+#ifdef _WIN32
+  replace(this->outputFile.begin(), this->outputFile.end(), '/', '\\');
+#endif
+
  if (this->fmkIn.empty()) {
    std::cerr << "INPUT MISSING: fmk is necessary";
    return RET_INPUT_PARAM_INVALID;