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的路径。

运行

1、数据读取

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

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

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

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

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

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

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

图片大小统一到1024

图片经过缩放并填充

5、输出预选矿的情况anchors

6、输出一个cell的anchor情况

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

正候选框(positive anchors)

负候选框(negative anchors)

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

8、ROI

猜你喜欢

转载自blog.csdn.net/Diana_Z/article/details/83578400