Skip to content

关于 dense relation distill module #25

@Saralyliu

Description

@Saralyliu
def forward(self,features,attentions):
    features = list(features)
    if isinstance(attentions,dict):
        for i in range(len(attentions)):
            if i==0:
                atten = attentions[i].unsqueeze(0)
            else:
                atten = torch.cat((atten,attentions[i].unsqueeze(0)),dim=0)
        attentions = atten.cuda()
    output = []
    h , w = attentions.shape[2:]
    ncls = attentions.shape[0]       
    key_t = self.key_t(attentions)   
    val_t = self.value_t(attentions) 
    for idx in range(len(features)):

我想知道谁是support ,谁是query?按代码的意思,是attentions是support,features是query是这样吗?
attentions和features的维度分别是多少?
期待您的回复,谢谢!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions