mrwu-mac / DIFNet

This repository is for the paper ``DIFNet: Boosting Visual Information Flow for Image Captioning'' (CVPR 2022).
BSD 3-Clause "New" or "Revised" License
20 stars 7 forks source link

关于分割特征提取的代码 #13

Open hzf-hhh opened 1 year ago

hzf-hhh commented 1 year ago

作者,您好,您方便提供一下具体提取分割特征的代码吗,我在UPSNet并没有找到相关的代码,期待您的回复

mrwu-mac commented 1 year ago

我们没有整理这部分代码,暂时无法提供,您可以使用我们提供的分割特征或者使用一些最新的分割框架(例如detectron2,只需要少量的修改即可)提取分割特征

hzf-hhh commented 1 year ago

我们没有整理这部分代码,暂时无法提供,您可以使用我们提供的分割特征或者使用一些最新的分割框架(例如detectron2,只需要少量的修改即可)提取分割特征

好的,谢谢您的回复,我想问下是直接利用分割模型的结果再做池化就可以得到分割特征吗

mrwu-mac commented 1 year ago

如果得到的是分割图,需要先转成bit图,之后再做平均池化即可

hzf-hhh commented 1 year ago

谢谢您的回复,您说的bit图是下面的这种灰度图吗,最后特征的133应该是语义分割的类别数目吧,我在平均池化的时候有点无从下手,您是如何把最后结果塑造成(133,7,7)的张量,您可以指点一下吗或者您有这部分对位图预处理的文件可以方便让我参考一下吗,我想在自己的图像上提取语义分割特征。 result4

mrwu-mac commented 1 year ago

抱歉未能及时回复,这里提供分割图转bit图伪代码,

result = zeros(class_num, h, w) 
for c in range(class_num):
    result[c] = segmap == c
hzf-hhh commented 1 year ago

抱歉未能及时回复,这里提供分割图转bit图伪代码,

result = zeros(class_num, h, w) 
for c in range(class_num):
    result[c] = segmap == c

非常感谢您的回复,祝您生活愉快

zdpaa commented 2 months ago

作者,您好,您方便提供一下具体提取分割特征的代码吗,我在UPSNet并没有找到相关的代码,期待您的回复

你好,你这个问题解决了吗?能提供下相关的代码用来提取分割特征嘛?

0xzzq666 commented 3 weeks ago

抱歉未能及时回复,这里提供分割图转bit图伪代码,

result = zeros(class_num, h, w) 
for c in range(class_num):
    result[c] = segmap == c

非常感谢您的回复,祝您生活愉快

您好,请问您的问题解决了吗?您方便提供一下具体提取分割特征的代码吗