- 博客(13)
- 收藏
- 关注
原创 coco实例分割mask用于语义分割
所以coco实力分割的mask转语义分割思路很简单,将同一个类别的实例mask直接相加,然后转成布尔类型即可。coco数据集的mask将每个实例都分离出来了,这通常适用于实例分割或者全景分割。对于语义分割而言,每一个mask包含每个类别的所有实例。
2023-07-02 02:36:15 644 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 630 1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人