def __init__(self, im=None, im_path=None):
self.stack = Stack()
if im is not None:
self.im = im
elif im_path is not None:
self.imread(im_path)
else:
raise ValueError("No Input image!")
self.im_size = self.im.shape
self.im_height, self.im_width = self.im_size
self.im_label = np.full_like(self.im, 0)
#import ipdb; ipdb.set_trace()
self.max_label = 0
def __init__(self, im=None, im_path=None):
self.stack = Stack()
if im is not None:
self.im = im
elif im_path is not None:
self.imread(im_path)
else:
raise ValueError("No Input image!")
self.im_size = self.im.shape
self.im_height, self.im_width = self.im_size
self.im_label = np.full_like(self.im, 0).astype(np.int)
self.max_label = 0
self.label_area = {}
