如何将matplotlib注释动作嵌入到函数中？

Question

使用matplotlib，您可以与打印的图像进行交互，例如进行注释。交互之后，您将获得一些通常要保存的数据。我想将所有内容合并到一个函数中，但是我不知道该怎么做。光谱模块说明了我的问题，但我之前遇到过。

以下是一些代码：

import spectral  as spy

class Annotator:
    def __init__(self, filename):
        self.filename = filename
        self.classes = list(zip(range(5), ['unknown','background','tape','healthy','infected']))
    def img(self): 
        img=spy.open_image(self.filename)
        return img
    def viewer(self):
        img = self.img()
        print(self.classes)
        viewer = spy.imshow(img)
        annotation = viewer.classes
        return annotation

如果我调用注释器An=Annotator(some data file)，然后调用查看器进行注释

results = An.viewer

然后python立即解析到最后，并给出一个空的none类型对象。我想破坏代码，以便我可以获取注释并将其保存在同一函数调用中。

所以我的问题是，我该怎么做？