def load_images(self):
self.left_pic = cv2.imread(self.left_file, 1)
self.right_pic = cv2.imread(self.right_file, 1)
shape = [
max(v1, v2) for v1, v2 in zip(self.left_pic.shape[:2],
self.right_pic.shape[:2])
]
self.left_img = self.left_axe.imshow(self.left_pic[::-1, :, ::-1],
origin="lower")
self.left_axe.axis("off")
self.left_mask = ImageMaskDrawer(self.left_axe,
self.left_img,
mask_shape=shape)
self.right_img = self.right_axe.imshow(self.right_pic[::-1, :, ::-1],
origin="lower")
self.right_axe.axis("off")
self.right_mask = ImageMaskDrawer(self.right_axe,
self.right_img,
mask_shape=shape)
self.left_mask.on_trait_change(self.mask_changed, "drawed")
self.right_mask.on_trait_change(self.mask_changed, "drawed")
self.figure.canvas.draw_idle()
def init_draw(self):
h, w = self.img.shape[:2]
if self.mask_artist is not None:
self.mask_artist.remove()
self.mask_artist = ImageMaskDrawer(self.axe, mask_shape=(h, w),
canmove=False, size=10)
self.mask_artist.on_trait_change(self.update_markers, "drawed")
self.markers = np.zeros((h, w), dtype=np.int32)
self.result = np.zeros((h, w), dtype=np.int32)
def init_draw(self):
self.mask_artist = ImageMaskDrawer(self.axe,
mask_shape=(512, 512),
canmove=False,
size=15)
self.connect_dirty("mask_artist.mask_updated")
def init_draw(self):
self.mask_artist = ImageMaskDrawer(self.axe,
self.img,
canmove=False,
size=15)
self.mask_artist.on_trait_change(self.draw, "drawed")
