diff --git a/labelImg.py b/labelImg.py index e926866c..a6516971 100755 --- a/labelImg.py +++ b/labelImg.py @@ -706,7 +706,7 @@ class MainWindow(QMainWindow, WindowMixin): ## Label xml file and show bound box according to its filename if self.usingPascalVocFormat is True and \ self.defaultSaveDir is not None: - basename = os.path.basename(self.filename).split('.')[:-1][0] + basename = os.path.basename(os.path.splitext(self.filename)[0]) xmlPath = os.path.join(self.defaultSaveDir, basename + '.xml') self.loadPascalXMLByFilename(xmlPath)