tradysun
tradysun
不要用convert_weight.py,直接运行freeze_weight.py,注意修改里面要生成pb文件的ckpt
 我加了4处改动,前两行是指定GPU卡号,你主要改pb_file(你要声明的pb名,自己起一个),ckpt_file(选一个loss结果最好的)
你算出的mAP是0.00%,说明没有预测出任何结果,从你图1的log结果可看出predict为空,应该是你训练的结果有问题,是不是用的ckpt loss太大,模型还没收敛。 另外也可以先做两个尝试: 1)修改下evaluate.py,trainable改为False  2)把core/config.py 中TEST的SCORE 和 THRESHOLD阈值改小一点,看看能不能出结果,如果输出的框乱七八糟,说明你的模型不好,建议检查数据,重新多训练几个epoch,待loss收敛较小为止
之前也有人遇到这个问题,ckpt转pb可以预测,但直接evaluate.py没有结果,试下把trainable改成False
> >  > > 我加了4處收縮,前兩行是指定GPU卡號,你主要改成pb_file(你要聲明的pb名,自己起一個),ckpt_file(選一個損失結果最好的) > > 想問如果用gpu訓練的話,是一定要打指定gpu那行嗎? 不是必须指定GPU卡的,但如果不指定,默认会把所有的卡内存占满,别人想用其他卡的话就会出现内存溢出,如果你只有一个卡就没这个问题。如果多个卡中有一个是坏的,就会报错,我就遇到这种问题。另外,作者的代码是不支持多GPU的,就算打满多个卡,也只是用第一个卡在训练。
你这个看着不像是报错,应该是linux下无法display照片,试试把image.show()改为image.save,不显示,保存下来
作者的代码好像还不支持多GPU训练
邮件已收到