出现错误的原因是输入的张量维度不对,需要将输入的张量从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')

这样就可以解决该错误。

MindSpore 人脸识别模型错误解决:输入维度问题

原文地址: https://www.cveoy.top/t/topic/jqkZ 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录