pytorch报错:RuntimeError: Given groups=1, weight of size 10 3 3 3, expected input[1, 12, 12, 3]
在mtcnn人脸检测中,网络训练时报以下错误:后来找到原因如下:图片经过处理后的数据格式是NHWC,而pytorch输入图片的格式要求是NCHW,需要转化一下。有两种方法:一是用下列方法:importtorchimage=torch.randn(1,12,12,3)image=image.permute(0,3,1,2)print(image.shape)#([1,3,12,12])二是用torc