tradysun

Results 8 comments of tradysun

不要用convert_weight.py,直接运行freeze_weight.py,注意修改里面要生成pb文件的ckpt

![image](https://user-images.githubusercontent.com/24379156/68747224-0f6c2380-0635-11ea-96d7-22a5812e2f57.png) 我加了4处改动,前两行是指定GPU卡号,你主要改pb_file(你要声明的pb名,自己起一个),ckpt_file(选一个loss结果最好的)

你算出的mAP是0.00%,说明没有预测出任何结果,从你图1的log结果可看出predict为空,应该是你训练的结果有问题,是不是用的ckpt loss太大,模型还没收敛。 另外也可以先做两个尝试: 1)修改下evaluate.py,trainable改为False ![image](https://user-images.githubusercontent.com/24379156/68832261-751ae700-06eb-11ea-8619-7ed85559fc0b.png) 2)把core/config.py 中TEST的SCORE 和 THRESHOLD阈值改小一点,看看能不能出结果,如果输出的框乱七八糟,说明你的模型不好,建议检查数据,重新多训练几个epoch,待loss收敛较小为止

之前也有人遇到这个问题,ckpt转pb可以预测,但直接evaluate.py没有结果,试下把trainable改成False

> > ![圖片](https://user-images.githubusercontent.com/24379156/68747224-0f6c2380-0635-11ea-96d7-22a5812e2f57.png) > > 我加了4處收縮,前兩行是指定GPU卡號,你主要改成pb_file(你要聲明的pb名,自己起一個),ckpt_file(選一個損失結果最好的) > > 想問如果用gpu訓練的話,是一定要打指定gpu那行嗎? 不是必须指定GPU卡的,但如果不指定,默认会把所有的卡内存占满,别人想用其他卡的话就会出现内存溢出,如果你只有一个卡就没这个问题。如果多个卡中有一个是坏的,就会报错,我就遇到这种问题。另外,作者的代码是不支持多GPU的,就算打满多个卡,也只是用第一个卡在训练。

你这个看着不像是报错,应该是linux下无法display照片,试试把image.show()改为image.save,不显示,保存下来

作者的代码好像还不支持多GPU训练