deformable DETR 核心思路记录

*小呆

于 2021-12-23 16:58:44 发布

阅读量3.5k

点赞数

分类专栏： deep learning 文章标签：深度学习计算机视觉 pytorch

本文链接：https://blog.csdn.net/qq_39575835/article/details/122090409

版权

原文链接：https://arxiv.org/abs/2010.04159

encoder_layer = MSDeformAttnTransformerEncoderLayer

MSDeformAttnTransformerEncoderLayer(
  (self_attn): MSDeformAttn(
    (sampling_offsets): Linear(in_features=256, out_features=192, bias=True)
    (attention_weights): Linear(in_features=256, out_features=96, bias=True)
    (value_proj): Linear(in_features=256, out_features=256, bias