您的位置:首页 > 产品设计 > UI/UE

ValueError: total size of new array must be unchanged

2017-08-16 11:47 633 查看
在对数据增强后的fasterrcnn中进行训练时,出现这个错误,原因是在lib/roi_data_layer/layer.py中,会出现

inds=np.reshape(inds,(-1,2))
因此有效的roidb需为偶数。
参考https://github.com/rbgirshick/py-faster-rcnn/issues/61可处理数据奇偶问题


#inds=np.reshape(inds,(-1,2))
#row_perm=np.random.permutation(np.arange(inds.shape[0]))
#inds=np.reshape(inds[row_perm,:],(-1,))

#considerthecasewhenlen(inds)isodd
isodd=len(inds)%2
ifnotisodd:
inds=np.reshape(inds,(-1,2))
else:
tail=inds[-1]
inds=np.reshape(inds[:-1],(-1,2))
row_perm=np.random.permutation(np.arange(inds.shape[0]))
inds=np.reshape(inds[row_perm,:],(-1,))
ifisodd:
inds=np.append(inds,tail)
self._perm=inds



                                            
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐