mask-RCNN笔记——inspect_data使用

mask-RCNN的使用,其中代码inspect_data的使用,用于dateset的读取

1、下载coco数据库,安装coco工具包pycocotools

https://blog.csdn.net/Diana_Z/article/details/83576598

2、新建一个*.py文件将*。ipynb的代码复制过去

3、数据路径修改:

修改代码中下图红框处COCO_DIR变量的值为数据文件夹dataset的路径。

mask-RCNN笔记——inspect_data使用_第1张图片

运行

1、数据读取

mask-RCNN笔记——inspect_data使用_第2张图片

2、随机显示四张图片以及图片类掩码

mask-RCNN笔记——inspect_data使用_第3张图片

3、随机一张图片显示掩码以及类选框,并输出信息

mask-RCNN笔记——inspect_data使用_第4张图片

image_id:表示图片在dataset的图片列表中的下标

image和mask是两张与原图一样大小的图片,image是原图,mask是叠加之后的掩码图片

class_ids:是一个3*1的dnarry,表示有3个选框分别对应三个类id

bbox:是一个3*4的dnarry,每行表示一个选框的位置

mask-RCNN笔记——inspect_data使用_第5张图片

4、随机一张图片显示resize后的图片以及分类和掩码,

图片大小统一到1024

图片经过缩放并填充

mask-RCNN笔记——inspect_data使用_第6张图片

mask-RCNN笔记——inspect_data使用_第7张图片

5、输出预选矿的情况anchors

mask-RCNN笔记——inspect_data使用_第8张图片

mask-RCNN笔记——inspect_data使用_第9张图片

6、输出一个cell的anchor情况

mask-RCNN笔记——inspect_data使用_第10张图片

mask-RCNN笔记——inspect_data使用_第11张图片

7、显示候选框以及最终的结果

mask-RCNN笔记——inspect_data使用_第12张图片

正候选框(positive anchors)

mask-RCNN笔记——inspect_data使用_第13张图片

负候选框(negative anchors)

mask-RCNN笔记——inspect_data使用_第14张图片

其他选框,没有加入训练中

mask-RCNN笔记——inspect_data使用_第15张图片

8、ROI

mask-RCNN笔记——inspect_data使用_第16张图片

mask-RCNN笔记——inspect_data使用_第17张图片

 

你可能感兴趣的:(mask-RCNN笔记——inspect_data使用)