detectron+caffe2+maskrcnn汇总

项目地址:1.tensorflow+keras 

2.Detectron

3.MODEL_ZOO

Detectron

Detectron 总结(一)

Detectron 总结(二)

Detectron 总结(三)

detectron 模型建立过程(rpn除外)

Mask RCNN(这里多为tensorflow+keras

Mask RCNN训练自己的数据集

TensorFlow实战:Chapter-8上(Mask R-CNN介绍与实现)(操作显示)

MaskRCNN源码解读(代码讲解)

【Mask R-CNN】代码专栏

『计算机视觉』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;

猜你喜欢

转载自blog.csdn.net/m0_37644085/article/details/85002979