怎么把分类的backbone改成提取特征
收藏
我想提取图片中的有效信息,想换个BackBone试试效果,但是看到paddleseg中内置的backbone 都是分类的,输出是(1,1000)这种,我想输出(1,3,112,112),这种,原始图片大小是(3,256,256),我不会改,只能把最后一次输出层删掉,然后加个上采样这种,但是效果很差,我想是不是不能这种改,这种乱改是不是破坏了原有的网络反而不能提取图片的特征。
0
收藏
请登录后评论
不要全连接层就好了,你看看我的pp-yolo项目就明白了。
去掉最后面的全连接层就行
语义分割领域就是这样用的,一般还会加空洞卷积增大感受野