Beispiel #1
0
class NoGuiRunner:
    def __init__(self, args):
        self._cw = CWrapper()
        self._args = args
        path = args.path

        self._grid = None
        self._image = None
        self._handles = None
        self.load_image(path)




    def run(self):
        self._grid = Grid(self._cw, self._image, self._args, gui=False)

        kpt_files = os.listdir(self._args.keypoints_dir)
        kpt_files.sort(key=lambda e: int(e.split('_')[0]))
        self._handles = self.add_bunch(os.path.join(self._args.keypoints_dir, kpt_files[0]))

        if not os.path.exists(self._args.output_dir):
            os.mkdir(self._args.output_dir)

        pose_num = 0
        cv2.imwrite(os.path.join(self._args.output_dir, '{}.png'.format(pose_num)),
                    self._image._data[:, :, ::-1])

        for pose in tqdm(kpt_files[1:]):
            self.move_bunch(pose)
            pose_num += 1
            for _ in range(self._args.num_iterations):
                self.run_once()
            cv2.imwrite(os.path.join(self._args.output_dir, '{}.png'.format(pose_num)),
                        self._image._data[:, :, ::-1])

    def run_once(self):
        # no time delay left
        self._grid.regularize()
        self._grid.project()



    def load_image(self, path):
            self._image = ImageHelper(self._cw, self._args, gui=False)

    def add_bunch(self, posepath):
        with open(posepath, 'r') as f:
            poss = json.load(f)
        kpts = poss['people'][0]['pose_keypoints_2d']
        xs = kpts[::3]
        ys = kpts[1::3]
        new_handles = {}
        i = 0
        for ptx, pty in zip(xs, ys):
            new_handles[i] = (ptx, pty)
            i += 1
        self._grid.create_bunch_cp(new_handles=new_handles)
        return new_handles

    def move_bunch(self, filename):
        with open(os.path.join(self._args.keypoints_dir, filename)) as f:
            poss = json.load(f)
        newpos = poss['people'][0]['pose_keypoints_2d']
        xs = newpos[::3]
        ys = newpos[1::3]
        for i, h_obj in self._handles.items():
            self._grid.set_control_target(i, xs[i], ys[i])
Beispiel #2
0
class Application:
    def __init__(self, path):
        self._cw = CWrapper()

        self._window = tk.Tk()

        self._grid = None
        self._image = None
        self.load_image(path)

        self._canvas = tk.Canvas(self._window,
                                 width=self._image.width,
                                 height=self._image.height)
        self._canvas.pack()

        self._image.canvas = self._canvas

        self._active_handle = -1
        self._loop = None
        self._t_last = 0

    def load_image(self, path):
        self._image = ImageHelper(self._cw, path)

    def bind(self, event, fn):
        self._canvas.bind(event, fn)

    def run(self):
        self._grid = Grid(self._cw, self._image)
        self._image.draw()
        self._grid.draw()

        self._run_once()

        self._window.mainloop()

    def _run_once(self):

        self._grid.regularize()

        dt = datetime.now()
        delta = dt.timestamp() - self._t_last
        if 0 < delta > 0.03:  # 0.03 - 30 FPS

            # dt = datetime.now()
            # t1 = dt.timestamp()

            self._grid.project()

            #dt = datetime.now()
            # print(dt.timestamp()-t1)

            self._image.draw()
            self._grid.draw()

            dt = datetime.now()
            self._t_last = dt.timestamp()

        self._loop = self._window.after(1, self._run_once)

    def select_handle(self, e):
        handle_id = self._image.select_handle(e.x, e.y)

        if handle_id == -1:
            handle_id = self._image.create_handle(e.x, e.y)
            if handle_id != -1:
                if not self._grid.create_control_point(handle_id, e.x, e.y):
                    self._image.remove_handle(handle_id)
                    return False
            else:
                return False

        self._active_handle = handle_id
        return True

    def deselect_handle(self, e):
        self._active_handle = -1

    def remove_handle(self, e):
        handle_id = self._image.select_handle(e.x, e.y)
        if handle_id != -1:
            self._grid.remove_control_point(handle_id)
            self._image.remove_handle(handle_id)

    def move_handle(self, e):
        if self._active_handle != -1:
            self._image.move_handle(self._active_handle, e.x, e.y)
            self._grid.set_control_target(self._active_handle, e.x, e.y)
Beispiel #3
0
class Application:

    def __init__(self, path):
        self._cw = CWrapper()

        self._window = tk.Tk()

        self._grid = None
        self._image = None
        self.load_image(path)

        self._canvas = tk.Canvas(self._window, width=self._image.width, height=self._image.height)
        self._canvas.pack()

        self._image.canvas = self._canvas

        self._active_handle = -1
        self._loop = None
        self._t_last = 0

    def load_image(self, path):
        self._image = ImageHelper(self._cw, path)

    def bind(self, event, fn):
        self._canvas.bind(event, fn)

    def run(self):
        self._grid = Grid(self._cw, self._image)
        self._image.draw()
        self._grid.draw()

        self._run_once()

        self._window.mainloop()

    def _run_once(self):

        self._grid.regularize()

        dt = datetime.now()
        delta = dt.timestamp()-self._t_last
        if 0 < delta > 0.03:  # 0.03 - 30 FPS

            # dt = datetime.now()
            # t1 = dt.timestamp()

            self._grid.project()

            #dt = datetime.now()
            # print(dt.timestamp()-t1)

            self._image.draw()
            self._grid.draw()

            dt = datetime.now()
            self._t_last = dt.timestamp()

        self._loop = self._window.after(1, self._run_once)

    def select_handle(self, e):
        handle_id = self._image.select_handle(e.x, e.y)

        if handle_id == -1:
            handle_id = self._image.create_handle(e.x, e.y)
            if handle_id != -1:
                if not self._grid.create_control_point(handle_id, e.x, e.y):
                    self._image.remove_handle(handle_id)
                    return False
            else:
                return False

        self._active_handle = handle_id
        return True

    def deselect_handle(self, e):
        self._active_handle = -1

    def remove_handle(self, e):
        handle_id = self._image.select_handle(e.x, e.y)
        if handle_id != -1:
            self._grid.remove_control_point(handle_id)
            self._image.remove_handle(handle_id)

    def move_handle(self, e):
        if self._active_handle != -1:
            self._image.move_handle(self._active_handle, e.x, e.y)
            self._grid.set_control_target(self._active_handle, e.x, e.y)