python-简单的resnet特征抽取器

阳光下的小白菜

已于 2022-01-22 17:06:00 修改

阅读量608

点赞数

分类专栏：多模态模型文章标签： python

于 2021-08-21 14:15:20 首次发布

本文链接：https://blog.csdn.net/kanon122500000/article/details/119839101

版权

本文介绍了如何使用Python实现ResNet模型作为特征抽取器。通过搭建ResNet网络并训练，可以提取图像数据的有效特征，为图像分类或其他计算机视觉任务提供预处理支持。

摘要由CSDN通过智能技术生成


class imageEncoder(nn.Module):
    def __init__(self, args):
        super().__init__()
        self.args = args
        self.resnet50 = cv_models.resnet50()
        #print(self.resnet50)
        load_resnet(args.cv_model_path, self.resnet50)
        # args.remove_relu = True