update mnnsr: detect tilesize & scale from model

Author: tumuyan

RealSR-NCNN-Android-CLI/MNN-SR/src/main/jni/main.cpp

@@ -6,6 +6,7 @@
 #include <clocale>
 #include <thread>
 #include <filesystem>
+#include <regex>
 
 //#undef min
 //#undef max
@@ -797,6 +798,37 @@ int main(int argc, char **argv)
         return -1;
     }
 
+    if (scale == 0) {
+#if _WIN32
+        using string_t = std::wstring;
+        using regex_t = std::wregex;
+        using smatch_t = std::wsmatch;
+#define STR(x) L##x
+#else
+        using string_t = std::string;
+        using regex_t = std::regex;
+        using smatch_t = std::smatch;
+#define STR(x) x
+#endif
+        // 获取文件名
+        string_t filename = std::filesystem::path(model).filename().native();
+
+        //regex_t re1(STR("(.+-|^)[xX]([0-9]+(\\.[0-9]+)?).*"));
+        //regex_t re2(STR("(.+-|^)([0-9]+(\\.[0-9]+)?)[xX].*"));
+        regex_t re1(STR("(.+-|^)[xX]([0-9]+).*"));
+        regex_t re2(STR("(.+-|^)([0-9]+)[xX].*"));
+
+        smatch_t match;
+        if (std::regex_search(filename, match, re1) && match.size() > 1) {
+            //scale = std::stod(match[2]);
+            scale = std::stoi(match[2]);
+        }
+        else if (std::regex_search(filename, match, re2) && match.size() > 1) {
+            //scale = std::stod(match[2]);
+            scale = std::stoi(match[2]);
+        }
+    }
+
 
 #include <cstdio>
 

RealSR-NCNN-Android-CLI/MNN-SR/src/main/jni/mnnsr.cpp

@@ -148,6 +148,18 @@ int MNNSR::load(const std::string &modelpath, bool cachemodel,const bool nchw)
 
 
     interpreter_input = interpreter->getSessionInput(session, nullptr);
+    auto dims = interpreter_input->shape();
+	if (dims.size() != 4) {
+		fprintf(stderr, "model input tensor shape error, expect 4 dims, but got %zu\n", dims.size());
+		return -1;
+	}
+	else if (dims[2] > 0 && dims[3] > 0 && dims[2] == dims[3]) {
+		if (dims[2] != tilesize) {
+			fprintf(stderr, "fix tilesize %d -> %d\n", tilesize, dims[2]);
+			tilesize = dims[2];
+		}
+	}
+
 //    fprintf(stderr, "model input tensor(b/c/h/w): %d/%d/%d/%d -> 1/%d/%d/%d\n"
 //            , input_tensor->batch(), input_tensor->channel(), input_tensor->height(), input_tensor->width()
 //            ,model_channel, tilesize, tilesize