网站首页
编程语言
数据库
系统相关
其他分享
编程问答
SFMA
2024-11-17
SFMA(提取全局和局部特征 并进行简单的融合)
importtorchimporttorch.nnasnnimporttorch.nn.functionalasFclassDMlp(nn.Module):'''用来提取局部特征'''def__init__(self,dim,growth_rate=2.0):super().__init__()hidden_dim=int(dim*