[DeepLearning] Mask RCNN 代码 tensorflow coco demo

源码地址

  1. 先按照demo.ipynb配置好各个参数
  2. model.detect(imges,verbose=1)进行模型测试,得到相应结果result,以N为检测到的实例数,图片大小为hw
    class_ids:1
    N,每个元素存储实例类名对应的编号
    masks一个三维数组,h*(w*N),以行展开,每一行包含相应元素属于N个实例中的bool值,实现像素级分割。
    rois:每个实例对应的bounding box,[N,[x1,y1,x2,y2]]
    scores:每个实例的置信分数。
  3. visualize.display_instances(image, r['rois'], r['masks'], r['class_ids'], class_names, r['scores'])
    np.where(condition, x, y) 满足条件(condition),输出x,不满足输出y
    为每一个实例制指定一种颜色。
    bounding box和mask都用这种颜色
    bounding box 用matplotlib.patches.Rectangle绘制
    mask的绘制很巧秒,见 apply_mask。
    mask polygon的绘制。利用skimage.measure.find_contours
    find_contours的x和y会相反相反一下,所以要转回来
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值