fast-rcnn训练时 snapshot保存训练模型时的问题

[hjl150505]

出错的是train.py里的snapshot函数： net.params['bbox_pred'][0].data[...] =
(net.params['bbox_pred'][0].data * self.bbox_stds[:, np.newaxis]) 我调试了下，net.params['bbox_pred'][0].data 它的shape是 8 * 32 * 1 * 1， self.bbox_stds[:, np.newaxis 它的shape是 8 * 1， 它俩相乘后的结果，shape是： 8 * 32 * 8 * 1 ，把这个结果保存到 net.params['bbox_pred'][0].data[...] 时会出错，因为net.params['bbox_pred'][0].data[...]的shape是 8 * 32 * 1 * 1，不知道你遇到过这个问题没？