diff --git a/waifu2x-caffe-gui/Source.cpp b/waifu2x-caffe-gui/Source.cpp index 60ec802..6399df7 100644 --- a/waifu2x-caffe-gui/Source.cpp +++ b/waifu2x-caffe-gui/Source.cpp @@ -766,7 +766,8 @@ public: std::string outputFileName = szTmp; const auto tailDot = outputFileName.find_last_of('.'); - outputFileName.erase(tailDot, outputFileName.length()); + if (tailDot != outputFileName.npos) + outputFileName.erase(tailDot, outputFileName.length()); const std::string addstr(AddName()); autoSetAddName = addstr;