yolov8 predict show=True - Iamge kills notebook kernel

Question

I am testing yolov8 prediction using the following code:

from ultralytics import YOLO

# Load a model

model = YOLO("yolov8n.pt")  

# Use the model

model.predict(source= "bus.jpg",show=True)  # predict on an image

This works perfectly in the Spyder IDE and the resulting image can be closed by clicking the toprighthand corner in the usual way.

Using the same code in a Jupter notebook also works but the image cannot be closed. Clicking on the image gives the popup message that Python is not responding, offering the choice of closing the program or waiting. AS expected clsing the program kills the kernel. Is there a way round this?

Answer 1

As this reminded me of using opencv and forgetting the cv2.destroyAllWindows a tried

cv2.waitKey(0)
cv2.destroyAllWindows() 

This works!