ROIAlign:ROIAlign来自Mask-RCNN

时间:2024-06-07 16:08:32
【文件属性】:

文件名称: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)

网友评论