python - Mask_RCNN。训练 Ballon.py 错误：使用形状分配张量时出现 OOM

问题描述

嘿，我已经按照以下教程使用 Mask RCNN tensorflow 进行对象检测：Mask RCNN github 嘿，当我在命令提示符下运行 Ballon.py train 命令“python balloon.py train --dataset=balloon1 --weights=coco”进行训练气球数据集我收到以下错误：命令提示错误我试图更改“Mask_RCNN\mrcnn\config.py”文件中的一些变量，但我仍然收到错误。有没有其他人遇到过这个问题？

标签： pythontensorflowkerasdeep-learningobject-detection

python - Mask_RCNN。训练 Ballon.py 错误：使用形状分配张量时出现 OOM

问题描述

解决方案

推荐阅读