项目地址:1.tensorflow+keras
Detectron
Mask RCNN(这里多为tensorflow+keras)
TensorFlow实战:Chapter-8上(Mask R-CNN介绍与实现)(操作显示)
MaskRCNN源码解读(代码讲解)
『计算机视觉』Mask-RCNN_推断以及预测网络分析(这个!)
caffe2系列
Caffe2 - (二十八)Detectron 之 modeling - model_builder
获取每一个图像Mask
def getMaskImg( mask,box,image ):
imageMask = image.copy();
for i in range(mask.shape[0]):
for j in range(mask.shape[1]):
if( mask[i,j]==0 ):
#imageMask[i,j] = image[i,j]
#mask[i,j] =254;
imageMask[i,j,0]=0;
imageMask[i,j,1]=0;
imageMask[i,j,2]=0;
x1 = int(box[0]);
x2 = int(box[2]);
y1 = int(box[1]);
y2 = int(box[3]);
imageMask = imageMask[y1:y2,x1:x2];
return imageMask;