should handle filenames with multiple periods

jveitchmichaelis · Jan 14, 2020 · 9420656 · 9420656
1 parent 54beddc
commit 9420656
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/src/cocoexporter.cpp b/src/cocoexporter.cpp
@@ -43,7 +43,7 @@ bool CocoExporter::processImages(const QString folder, const QString label_filen
         if(!export_unlabelled && labels.size() == 0) continue;
 
         QString extension = QFileInfo(image_path).suffix();
-        QString filename_noext = QFileInfo(image_path).baseName();
+        QString filename_noext = QFileInfo(image_path).completeBaseName();
 
         QString image_filename = QString("%1/%2.%3").arg(folder).arg(filename_noext).arg(extension);
 

diff --git a/src/darknetexporter.cpp b/src/darknetexporter.cpp
@@ -80,7 +80,7 @@ bool DarknetExporter::processImages(const QString folder, const QList<QString> i
         if(!export_unlabelled && labels.size() == 0) continue;
 
         QString extension = QFileInfo(image_path).suffix();
-        QString filename_noext = QFileInfo(image_path).baseName();
+        QString filename_noext = QFileInfo(image_path).completeBaseName();
         QString image_filename = QString("%1/%2.%3").arg(folder).arg(filename_noext).arg(extension);
 
         // Correct for duplicate file names in output

diff --git a/src/kittiexporter.cpp b/src/kittiexporter.cpp
@@ -69,7 +69,7 @@ int KittiExporter::processSet(QString folder, QList<QString> images, int i){
         if(!export_unlabelled && labels.size() == 0) continue;
 
         QString extension = QFileInfo(image).suffix();
-        QString filename_noext = QFileInfo(image).baseName();
+        QString filename_noext = QFileInfo(image).completeBaseName();
 
         QString image_filename = QString("%1/%2.%3").arg(folder).arg(filename_noext).arg(extension);
 

diff --git a/src/labelproject.cpp b/src/labelproject.cpp
@@ -265,7 +265,7 @@ void LabelProject::cancelLoad(){
 void LabelProject::addVideo(QString fileName, QString outputFolder){
     cv::VideoCapture video(fileName.toStdString());
 
-    QString base_name = QFileInfo(fileName).baseName();
+    QString base_name = QFileInfo(fileName).completeBaseName();
     QDir dir(outputFolder);
 
     cv::Mat frame;

diff --git a/src/pascalvocexporter.cpp b/src/pascalvocexporter.cpp
@@ -160,7 +160,7 @@ bool PascalVocExporter::processImages(const QString folder, const QList<QString>
 
 
         QString extension = QFileInfo(image_path).suffix();
-        QString filename_noext = QFileInfo(image_path).baseName();
+        QString filename_noext = QFileInfo(image_path).completeBaseName();
         QString image_filename = QString("%1/%2.%3").arg(folder).arg(filename_noext).arg(extension);
 
         // Correct for duplicate file names in output