修改Caffe VGG 16以在PyTorch上处理1个通道图像

作者: star
发布时间: 2025-03-05 08:26:21 (5天前)
转自：

2 条回复

0#
回复此人
一只特立独行的猪 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 您需要输入形状为Batch-Channel-Height-Width，即4D。在你的情况下，你只有一个通道，所以你“挤出”这个单一维度，但是pytorch不喜欢它！ </p> <P> 尝试 </p> <pre> <code> im2arr = im2arr[np.newaxis, np.newaxis, :, :] # add singleton for the channles as well </code> </pre> </DIV>

编辑

登录后才能参与评论