Pytorch и Matplotlib мешают ⇐ Python

[Anonymous]

Я столкнулся со странной ошибкой в ​​Matplotlib и torch. Если я запущу эту строку torch.hub.load, plt.imshow просто ничего не отобразит (даже если кадр является правильным изображением). Если я прокомментирую эту строку, plt.imshow будет работать.
Независимо от того, прокомментирована ли эта строка torch.hub.load или нет, cv2.imshow будет работать.

Код: Выделить всё

onnx_path = "my_weights.onnx"
yolo_path = "lib/yolov5/"

torch.hub.load(yolo_path, 'custom', path=onnx_path, source='local')

video_reader = VideoReader(str(src_file))

# wait for thread to read
while not video_reader.is_ready():
waiting += 1
time.sleep(1)

while(video_reader.is_ready()):
frame = video_reader.frame

#cv2.imshow('image',frame)
#cv2.waitKey(0)

plt.imshow(frame)
plt.axis('off')
plt.show()

Кажется, я что-то упускаю, но я этого не вижу. Любая помощь приветствуется

Подробнее здесь: https://stackoverflow.com/questions/783 ... nterfering