Mask_RCNN。训练Ballon.py错误:分配具有形状的张量时出现OOM

时间:2018-09-23 02:21:46

标签: python tensorflow keras deep-learning object-detection

嘿,我已经按照以下模板进行对象检测教程了:Mask RCNN tensorflow on:Mask RCNN github 嘿,当我在命令提示符下运行Ballon.py火车命令“ python Balloon.py train --dataset = balloon1 --weights = coco”以训练气球数据集时,出现以下错误: Command prompt error 我试图更改“ Mask_RCNN \ mrcnn \ config.py”文件中的某些变量,但仍然出现错误。还有其他人遇到过这个问题吗?

0 个答案:

没有答案