facePath = imagePath + "/" + img face = Image.open(facePath) imgblob = dataTransforms(face).unsqueeze(0) imgblob = Variable(imgblob) torch.no_grad() predict = F.softmax(net(imgblob), dim=1) predict是一个长度为2的一维张量,正常情况下是长度为1的一维张量。 这是我的邮箱,1658550916@qq.com。需要的话,我可以将这样的图片发给您。
predict是一个长度为2的一维张量,正常情况下是长度为1的一维张量。
这是我的邮箱,1658550916@qq.com。需要的话,我可以将这样的图片发给您。