interactivedatasetlabeling.py 文件源码

def saveCurrent(self):

        self._seq.data[self.curFrame] = self._seq.data[self.curFrame]._replace(gtorig=self.curData)
        if hasattr(self._seq.data[self.curFrame], 'extraData'):
            ed = {'pb': {'pbp': self.curPbP, 'pb': self.curPb}, 'vis': self.curVis}
            self._seq.data[self.curFrame] = self._seq.data[self.curFrame]._replace(extraData=ed)

        if self.filename_joints is not None and self.filename_pb is not None and self.filename_vis is not None:
            self.importer.saveSequenceAnnotations(self._seq, {'joints': self.filename_joints,
                                                              'vis': self.filename_vis,
                                                              'pb': self.filename_pb})

        # save log file
        if self.file_log is not None:
            self.file_log.write('{}, {}, {}@{}, {}\n'.format(self._seq.data[self.curFrame].fileName,
                                                             time.strftime("%d.%m.%Y %H:%M:%S +0000", time.gmtime()),
                                                             getpass.getuser(),
                                                             os.uname()[1],
                                                             time.time() - self.start_time))
            self.file_log.flush()