毛毛的博客

原创 CV_Task3 利用ResNet实现人脸识别

作业数据集描述：共15个人的头像，每个人11张图。作业描述：使用给定的数据集完成人脸识别。实现方法对人脸图像进行分类处理，可以使用目前流行的深度学习算法进行处理。ResNet作为经典的分类CNN模型，在分类问题上有着出色的表现，因此针对人脸分类问题可以使用ResNet模型作为分类网络来解决。在Pytorch中已经包含了预设的resnet模型可以直接使用。导入包from torchvision.models import resnet18import torchimport torch.nn