我在跑以下代码时出现错误
pic = pic[:,:, ::-1,:]
pic = torch.from_numpy(pic.transpose([3,0,1,2]))
## ValueError: some of the strides of a given numpy array are negative
解决方案
pic = pic[:,:, ::-1,:].copy() #加上.copy()
具体可参考:discuss.pytorch
本文介绍在使用PyTorch进行图像处理时遇到的“some of the strides of a given numpy array are negative”错误,并提供了一个简单有效的解决方案,即通过使用.copy()方法来避免负步长问题。
我在跑以下代码时出现错误
pic = pic[:,:, ::-1,:]
pic = torch.from_numpy(pic.transpose([3,0,1,2]))
## ValueError: some of the strides of a given numpy array are negative
解决方案
pic = pic[:,:, ::-1,:].copy() #加上.copy()
具体可参考:discuss.pytorch
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
704
2277
571

被折叠的 条评论
为什么被折叠?