tensorflow - Python内核在加载预训练的YOLO模型时死亡

Question

我试图在 spyder IDE 上使用网络摄像头实时提要运行 coursera 的 YOLO 对象检测实现。但是，每当它遇到：

yolo_model =load_model("model_data/yolo.h5")

内核死亡。

显示的错误是：

2018 15:18:14.079164: I C:\tf_jenkins\workspace\rel‑win\M\windows\PY\35\tensorflow\core\platform\cpu_feature_guard.cc:140] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2

标签： tensorflowkerasdeep-learningcomputer-visionyolo

Answer 1

当我在刚刚加载的 python shell 中运行那一行（以及之前的导入语句）时，我也会遇到同样的错误。模型文件为 200 MB。我有一台 16 GB RAM Windows 笔记本电脑，配备 GTX 1050 GPU 和 4GB VRAM。我还尝试通过我的 CPU 运行它（首先启动一个将设备设置为 cpu 的会话）。

我只能猜测 Coursera 的计算机比我有更多的内存。