facebook的timesformer代码的提取特征再哪个文件中呀,是在哪一步提取特征,还有全连接层
在 Facebook 的 Timm 库中,Timesformer 模型的特征提取代码位于 timm/models/vision_transformer.py 文件中的 T2T 类中的 forward_features 函数中。具体来说,通过调用 T2T 类实例的 forward_features 方法,可以获取到经过 Timesformer 模型提取后的特征。
而全连接层(即分类头)的代码位于 timm/models/vision_transformer.py 文件中的 T2T 类中的 head 属性中。在 Timesformer 模型中,分类头是由一个全局平均池化层和一个线性层组成,可以通过访问 T2T 实例的 head 属性来获取分类头。