文件名称:ROIAlign:ROIAlign来自Mask-RCNN
文件大小:4KB
文件格式:ZIP
更新时间:2024-06-07 16:08:32
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)