bd旋转验证码识别训练带模型

转验证码识别概述

这次我将带着大家实现旋转验证码的识别，当然当前时间点上识别教程有很多，我的教程会有自己优化的点(主要在损失函数上)，使准确率更加高，需要的数据集更加少。

项目开源地址

https://github.com/2833844911/Rotate-the-verification-code-pytorch/

定义模型

下面我们使用的是resnet50模型进行分类

class CustomResNet50(nn.Module):
    def __init__(self, num_classes=num_classes):
        super(CustomResNet50, self).__init__()
        # 加载ResNet50模型
        resnet50 = models.resnet50(pretrained=True)
        self.resnet_layers = nn.Sequential(*list(resnet50.children())[:-1]