YOLOV8对于classify分类任务更换resnet18主干网络

最新推荐文章于 2025-06-12 16:20:11 发布

西柚与蓝莓

最新推荐文章于 2025-06-12 16:20:11 发布

阅读量898

点赞数 7

CC 4.0 BY-SA版权

分类专栏： yolo 机器学习 python 文章标签： YOLO 分类

本文链接：https://blog.csdn.net/m0_53291740/article/details/141254340

1. 先定义残差18模块的网络

class Resnet18(nn.Module):
    def __init__(self):
        super().__init__()
        model = models.resnet18(pretrained=True)
        self.layer=nn.Sequential(
        model.conv1,
        model.bn1,
        model.relu,
        model.maxpool,
        model.layer1,
        model.layer2,
        model.layer3,
        model.layer4,
        model.avgpool
        )
    def forward(self, x):
        x=self.layer(x)
        return x

添加到conv.py末尾