【Segment Anything Model】十(2):利用SAM做分类任务,方法二,SAM的image_encoder部分进行特,将SAM提取到的ImageEmbedding插入到vgg网络中

博主微信 cvxiayixiao
【Segment Anything Model】计算机视觉检测分割任务专栏。 链接
【公开数据集预处理】特别是医疗公开数据集的接受和预处理,提供代码讲解。链接
【opencv+图像处理】opencv代码库讲解,结合图像处理知识,不仅仅是调库。链接

本篇分为两小节:

【Segment Anything Model】九(1)介绍加载vgg16网络的预训练权重,并利用其主干部分进行特征提取,改变分类个数为二值分类,或者自定义num_class分类。 【Segment Anything Model】九(2)介绍使用SAM的image_encoder部分进行特征提取,将提取到的Embedding插入到vgg16中,形成cnn和transform的混合特征,之后进行二分类。

本篇给出vgg和SAM联动代码。本篇和

你可能感兴趣的:(seg模型,分类,SAM,开源项目有代码,人工智能,特征融合)