Keras中的微小Yolov2内存消耗(tensorflow作为后端)

时间:2019-06-05 13:53:30

标签: tensorflow keras yolo

我正在尝试使用keras和tensorflow在我的机器上测试yolo tiny v2,但是我有内存警告。 警告说:“分配器试图找到2.29GiB时内存不足。调用方表明这不是故障,但可能意味着如果有更多可用内存,则可能会提高性能。”但是,看起来有2.29GB的GPU内存对于微型yolo型号来说太大了。我正在将网络应用到摄像头的视频流中。我有2GB GPU Quadro K1000M。

0 个答案:

没有答案