panel_group.py 文件源码

def OnChooseFolder(self, evt):
        """Choose Folder."""
        dlg = wx.DirDialog(self)
        if dlg.ShowModal() == wx.ID_OK:
            path = dlg.GetPath()
            del self.face_paths[:]
            self.faces.clear()
            for root, dirs, files in os.walk(path):
                if files:
                    self.face_paths.extend([
                        os.path.join(root, filename)
                        for filename in files
                    ])

            self.btn.Disable()

            self.log.log(
                'Request: Preparing faces for grouping, detecting faces in '
                'chosen folder.'
            )
            self.grid.set_paths(self.face_paths)
            for path in self.face_paths:
                try:
                    res = util.CF.face.detect(path)
                except util.CF.CognitiveFaceException:
                    continue
                for entry in res:
                    face = model.Face(entry, path)
                    self.faces[face.id] = face
            self.grid.set_faces(self.faces.values())
            self.log.log(
                'Response: Success. Total {0} faces are detected.'.format(
                    len(self.faces)))

            self.log.log('Request: Grouping {0} faces.'.format(
                len(self.faces)))
            res = util.CF.face.group(self.faces.keys())
            self.result.set_data(self.faces, res)
            len_groups = len(res['groups'])
            if res.get('messyGroup'):
                len_groups += 1
            self.log.log(
                'Response: Success. {0} faces grouped into {1} groups'.format(
                    len(self.faces), len_groups))

            self.btn.Enable()