MindSpore 人脸识别模型错误解决:输入维度问题
出现错误的原因是输入的张量维度不对,需要将输入的张量从5维转换为4维。可以在代码中添加以下语句来实现:
# 将输入的张量从5维转换为4维
face = np.squeeze(face, axis=0)
另外,还需要将模型的输入维度修改为4维,可以在ResNet模型的构造函数中添加以下语句:
self.conv1 = nn.Conv2d(3, 64, kernel_size=7, stride=2, padding=3, pad_mode='pad', has_bias=False, padding_mode='same')
这样就可以解决该错误。
原文地址: https://www.cveoy.top/t/topic/jqkZ 著作权归作者所有。请勿转载和采集!