summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authoryanchen036 <yanchen036@gmail.com>2017-01-02 05:33:49 (GMT)
committeryanchen036 <yanchen036@gmail.com>2018-02-15 07:21:45 (GMT)
commitb3deb95e75f6890c08e7aa60193ea304614b24ea (patch)
treea6007f02726ede2ff8208593394823484a9addc6
parenta44c444ee4ae0e7c0aa77118213d34bb26e9f2e6 (diff)
downloadcaffe-b3deb95e75f6890c08e7aa60193ea304614b24ea.zip
caffe-b3deb95e75f6890c08e7aa60193ea304614b24ea.tar.gz
caffe-b3deb95e75f6890c08e7aa60193ea304614b24ea.tar.bz2
bug fix: ext should not include the '.'
-rw-r--r--src/caffe/util/io.cpp2
-rw-r--r--tools/convert_imageset.cpp2
2 files changed, 2 insertions, 2 deletions
diff --git a/src/caffe/util/io.cpp b/src/caffe/util/io.cpp
index 835d2d4..5295d9d 100644
--- a/src/caffe/util/io.cpp
+++ b/src/caffe/util/io.cpp
@@ -106,7 +106,7 @@ cv::Mat ReadImageToCVMat(const string& filename) {
static bool matchExt(const std::string & fn,
std::string en) {
size_t p = fn.rfind('.');
- std::string ext = p != fn.npos ? fn.substr(p) : fn;
+ std::string ext = p != fn.npos ? fn.substr(p+1) : fn;
std::transform(ext.begin(), ext.end(), ext.begin(), ::tolower);
std::transform(en.begin(), en.end(), en.begin(), ::tolower);
if ( ext == en )
diff --git a/tools/convert_imageset.cpp b/tools/convert_imageset.cpp
index 90cdb15..9c5d09f 100644
--- a/tools/convert_imageset.cpp
+++ b/tools/convert_imageset.cpp
@@ -115,7 +115,7 @@ int main(int argc, char** argv) {
size_t p = fn.rfind('.');
if ( p == fn.npos )
LOG(WARNING) << "Failed to guess the encoding of '" << fn << "'";
- enc = fn.substr(p);
+ enc = fn.substr(p+1);
std::transform(enc.begin(), enc.end(), enc.begin(), ::tolower);
}
status = ReadImageToDatum(root_folder + lines[line_id].first,