def write_xml(self): imgFolderName = ustr(self.image_path).split('/')[-2] print imgFolderName #imgFileName = str(self.image_path).split('/')[-1] imgFileNameWithoutExt = ustr(self.image_path).split('.')[0] print imgFileNameWithoutExt image = QtGui.QImage() image.load(ustr(self.image_path)) imageShape = [ image.height(), image.width(), 1 if image.isGrayscale() else 3 ] writer = PascalVocWriter(imgFolderName, imgFileNameWithoutExt, imageShape, localImgPath=ustr(self.image_path)) #writer.verified = self.verified item = {} item['name'] = '' item['flood1name'] = imgFileNameWithoutExt + '_index.bmp' item['flood2name'] = imgFileNameWithoutExt + '_category.bmp' writer.addObject(item['name'], item['flood1name'], item['flood2name']) writer.save() return