改了新网络之后,能够训练但是测试时无法显示出图片,评估时报错:
BB = BB[sorted_ind, :]
IndexError: too many indices for array
暂时判断是因为测试网络没有改,或者训练网络本身就有问题,没有学到任何东西,但是loss降得很低了。
尝试第一种解决办法:
1,不使用预训练模型:在.sh文件里去掉
"--weights data/imagenet_models/${NET}.v2.caffemodel "
2,在卷积层和全连接层添加
weight_filler { type: "gaussian" std: 0.01 }
bias_filler