def images_to_single_fruit_centroid(fruit_number, image_path): import PIL image = PIL.Image.open(image_path).resize((320,240), PIL.Image.NEAREST) # fruit = (Image(image_path, grey=True)==fruit_number)*1.0 fruit = Image(image)==fruit_number # import pdb; pdb.set_trace() blobs = fruit.blobs() # plt.imshow(fruit.image) # plt.annotate(str(fruit_number), np.array(blobs[0].centroid).reshape(2,)) # plt.show() # assert len(blobs) == 1, "Must have only 1 fruit of the given fruit type in this photo" return np.array(blobs[0].centroid).reshape(2,) # (2,)
>>> im = Image('shark2.png') >>> blobs = im.blobs() >>> print(blobs) """ return Blob(self, **kwargs) if __name__ == "__main__": from machinevisiontoolbox import Image import matplotlib.pyplot as plt im = Image('shark2.png') blobs = im.blobs() print(blobs) im.disp() blobs.plot_labelbox(color='yellow') blobs.plot_centroid() plt.show() # # read image # from machinevisiontoolbox import Image # im = Image(cv.imread('images/multiblobs.png', cv.IMREAD_GRAYSCALE)) # # call Blobs class # b = Blob(image=im) # # plot image # # plot centroids of blobs