这个专栏更新各种AI,以及各种有趣的教程,有兴趣的小伙伴可以订阅一下。
直接上图,本期是检测人脸表情。
原理其实很简单,就是通过目标检测将人脸检测出来,
然后将检测出来的人脸用卷积网络训练的模型进行表情分类,达到表情识别的功能。
class LeNet(nn.Module):
def __init__(self, num_classes=num_classes):
super(LeNet, self).__init__()
self.conv1 = nn.Sequential( # input_size=(1*64*64)
nn.Conv2d(3, 6, 5, 1, 2), # padding=2
nn.ReLU(), # input_size=(6*64*64)
nn.MaxPool2d(kernel_size=2, stride=2), # output_size=(6*32*32)
)
self.conv2 = nn.Sequential(
nn.Conv2d(6, 16, 5),