Spaces:
Runtime error
Runtime error
| import cv2 | |
| from lib.MiVOS_STCN.interact.interaction import ScribbleInteraction | |
| class MyScribbleInteraction(ScribbleInteraction): | |
| def __init__(self, image, prev_mask, true_size, controller, num_objects): | |
| super().__init__(image, prev_mask, true_size, controller, num_objects) | |
| def push_drawing(self, drawing_points, k=1): | |
| """ | |
| Saves a drawing for the scribble to mask interaction | |
| :param drawing_points: Points used in the drawing, without connection lines | |
| :param k: Object id, 0 for negative scribbles, 1 else | |
| """ | |
| self.curr_path[k] = drawing_points | |
| selected = self.curr_path[k] | |
| for i in range(len(selected) - 1): | |
| cv2.line(self.drawn_map, selected[i], selected[i + 1], k, thickness=self.size) | |
| def scale_points(points, h1, w1, h2, w2): | |
| """ | |
| Scaling points in a list from (h1, w1) to (h2, w2) | |
| :param points: | |
| :param h1: Scaling from this height | |
| :param w1: Scaling from this width | |
| :param h2: Scaling to this height | |
| :param w2: Scaling to this width | |
| :return: List of scaled points | |
| """ | |
| h_scale = h2 / h1 | |
| w_scale = w2 / w1 | |
| res = [] | |
| for x, y in points: | |
| scaled_x = int(x * w_scale) | |
| scaled_y = int(y * h_scale) | |
| res.append((scaled_x, scaled_y)) | |
| return res | |