- 博客(2)
- 收藏
- 关注
RSS订阅原创 Yolact运行内存报错
然后发现报错的数组大小不合理，远大于我的图片大小，debug找原因，发现340是我的单张图片的目标数，没办法就这么多，图像尺寸却比我的图片大好2 3倍，这就是导致内存爆炸的原因。本来以为是模型参数设置不合理，导致运行内存爆炸，但在设置num_works = 0 batchsize=1的条件后问题依然存在。解决办法：找到expand类，其是用来扩展图片大小的，修改参数ratio为1到1.5。
2024-04-07 22:05:48
112
原创 Mask R-CNN 理解记录
将PRN的输出结果，在特征图上进行裁剪，就是进行roialign操作，实际是经过线性插值生成统一的尺寸7*7*c，然后一个分支经过两次FCN操作生成框预测头，用于进行类别损失和框几何损失计算，另一个分支经过FPN操作生成m*m*k的结果，k为类别数，最后需要进行阈值分割确定前景和背景。PRN输入的就是主干网络输出的特征层，假设为c*h*w,经过一次3*3卷积操作256*h*w，然后经过1*1卷积后得到两个分支的结果：2k*h*w和4k*h*w，通过阈值还有边界条件等删除一些框。
2024-04-06 21:32:23
316
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人