![](https://img-blog.csdnimg.cn/8b65b3c142714579beb7c7e4eb402c61.jpeg?x-oss-process=image/resize,m_fixed,h_224,w_224)
计算机视觉
Computer Vision的研究
尼古拉斯喵呜
大四/炼丹/GitHub: https://github.com/Elm-Forest
展开
-
coco实例分割mask用于语义分割
所以coco实力分割的mask转语义分割思路很简单,将同一个类别的实例mask直接相加,然后转成布尔类型即可。coco数据集的mask将每个实例都分离出来了,这通常适用于实例分割或者全景分割。对于语义分割而言,每一个mask包含每个类别的所有实例。原创 2023-07-02 02:36:15 · 567 阅读 · 1 评论 -
关于MultiheadAttention的权重
最近在做DETR相关的work,因为要做注意力权重可视化,在加载了DETR的官方权重后,并未能找到wq,wk,wv类似的权重参数,仅仅看到这些参数。q , k , v的具体计算,在nn.functional的_scaled_dot_product_attention方法里。通过解析nn.MultiheadAttention具体的实现,还是颇有收获。原方法写的很繁杂,这里贴一些具有代表性的部分。得到q, k, v,在nn.functional的_in_projection_packed方法里实现。原创 2022-12-15 20:40:24 · 584 阅读 · 1 评论