Не могу выйти из окна matplotlib

Question

У меня есть этот код, который должен нарисовать прямоугольник в окне matplotlib, а затем распечатать координаты этого изображения, у меня также есть его, так что когда я нажимаю 'q', он переходит к следующему изображению, однако этоне работаетНо я могу нажать кнопку x в окне, и оно перейдет к следующему изображению, но не выведет координаты

import os
import matplotlib.pyplot as plt
import cv2
from matplotlib.widgets import RectangleSelector

img = None
tl_list = []
br_list = []
object_list = []

image_folder = 'images'
savedir = 'annotations'
obj = 'thumbs up'

def line_select_callback(clk, rls):
    global tl_list
    global br_list
    tl_list.append((int(clk.xdata), int(clk.ydata)))
    br_list.append((int(rls.xdata), int(rls.ydata)))

def toggle_selector(event):
    toggle_selector.RS.set_active(True)

def onkeypress(event):
    global tl_list
    global br_list
    global object_list
    global img
    if event.key == 'q':
        print(tl_list, br_list)
        tl_list = []
        br_list = []
        img = None
        plt.close()

if __name__ == '__main__':
    for n, image_file in enumerate(os.scandir(image_folder)):
        img = image_filewwwwwetgfdsx
        fig, ax = plt.subplots(1)
        image = cv2.imread(image_file.path)
        image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
        ax.imshow(image)

        toggle_selector.RS = RectangleSelector(
            ax, line_select_callback,
            drawtype='box', useblit =True,
            button=[1], # left mouse click
            minspanx=5, minspany=5, spancoords='pixels',
            interactive=True
        )
        bbox = plt.connect('key_press_event', toggle_selector)
        key = plt.connect('key_press_event', onkeypress)
        plt.show()

Answer 1

Извините всех, мне просто пришлось обновить Matplotlib, этот код работает только в 3.0, а не в 2.0.