1 http://www.cnblogs.com/objectDetect/p/5780006.html
2 http://blog.csdn.net/samylee/article/details/51822832
在按照2训练时,运行 python examples/ssd/ssd_pascal.py, 到Iteration0时Ubuntu就重启了,
可能是显卡计算量有限,把ssd_pascal.py里面的 batch_size = 32改为batch_size = 16就ok了
(注意,上面是有问题的,每次迭代都是 loss=nan,说明是大问题,不要继续往下跑了,赶紧查查看是哪里的问题)
这样才对,loss不能为nan 的。
迭代60000次, 耗时1天,训练完毕:
上面这个图是有问题的,loss=nan,下面的图是正确的:
看结果:
再运行: python examples/ssd/score_ssd_pascal.py
看到detection的数值0.719106,跟教程里面的一致