Exemplo n.º 1
0
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