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