support the scientific notation when parsing layer's paramters.

Tencent · Apr 23, 2018 · 89c7aa2 · 89c7aa2
1 parent d0bcda7
commit 89c7aa2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/paramdict.cpp b/src/paramdict.cpp
@@ -12,6 +12,7 @@
 // CONDITIONS OF ANY KIND, either express or implied. See the License for the
 // specific language governing permissions and limitations under the License.
 
+#include <ctype.h>
 #include "paramdict.h"
 #include "platform.h"
 
@@ -74,7 +75,7 @@ static bool vstr_is_float(const char vstr[16])
         if (vstr[j] == '\0')
             break;
 
-        if (vstr[j] == '.')
+        if (vstr[j] == '.' || tolower(vstr[j]) == 'e')
             return true;
     }
 

diff --git a/tools/ncnn2mem.cpp b/tools/ncnn2mem.cpp
@@ -67,7 +67,7 @@ static bool vstr_is_float(const char vstr[16])
         if (vstr[j] == '\0')
             break;
 
-        if (vstr[j] == '.')
+        if (vstr[j] == '.' || tolower(vstr[j]) == 'e')
             return true;
     }