【文件属性】:
文件名称:ROIAlign:ROIAlign来自Mask-RCNN
文件大小:4KB
文件格式:ZIP
更新时间:2021-05-18 22:21:52
Cuda
ROIAlign
ROIAlign来自Mask-RCNN
这是Mask-RCNN的ROIAlign的体现: ://arxiv.org/abs/1703.06870。 考虑到matconvnet( )中没有ROIAlign,我只是写了这句话,以便我们可以在matlab中使用roialign层。
该代码以CUDA(.cu)和mex文件编写,您可以在带有GPU的matlab中使用它。 在matlab中使用命令:
mexcuda roialign.cu
编译文件。
然后您应该将这些代码添加到matconvnet中的vl_simplenn.m
case 'roialign'
res(i+ 1 ).x = roialign(res(i).x,rois,l.subdivisions,l.transform);
这是前向传播
case 'roialign'
res(i).
【文件预览】:
ROIAlign-master
----roialign.cu(13KB)
----README.md(785B)
----.gitignore(38B)