- 先按照demo.ipynb配置好各个参数
model.detect(imges,verbose=1)
进行模型测试,得到相应结果result
,以N为检测到的实例数,图片大小为hw
class_ids
:1N,每个元素存储实例类名对应的编号
masks
一个三维数组,h*(w*N),以行展开,每一行包含相应元素属于N个实例中的bool值,实现像素级分割。
rois
:每个实例对应的bounding box,[N,[x1,y1,x2,y2]]
scores
:每个实例的置信分数。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会相反相反一下,所以要转回来
[DeepLearning] Mask RCNN 代码 tensorflow coco demo
最新推荐文章于 2024-10-06 17:15:42 发布