windows+tensorflow2+python3环境配置mask-rcnn_v2.1详解
tf2改版后有了较大的改动,以前很多的经典算法都建立在tf1的版本上,需要做相应的修改才能在tf2上运行。本文详细记录了在win+tf2+python3环境下配置mask-rcnn_v2.1气球检测版本的过程以及遇到的问题。本机具体版本为windows10,python3.7,tensorflow2.1.0。
Github下载源码
下载GIthub源码与数据
首先搜索适用于win+tf的mask-rcnn源码,选择一个星星比较多的版本,很明显,我们选第一个。
进入第一个项目,先大概浏览一下,着重阅读Installation内容,大致了解项目的配置使用流程。
点击releases page,进入发布的新版本以及预训练模型参数页面。
依次下载气球数据集balloon_dataset.zip、预训练模型参数mask_rcnn_balloon.h5(也可以下载coco数据集上的预训练模型参数mask_rcnn_coco.h5做fine-tuning,本文下载了后者fune-tune模型)、源代码Source code(zip)或Source code(tar.gz)均可。
下载后的文件如图:
解压文件放置到对应目录
1.先解压v2.1文件,随便命名为自己想要的名字,解压后文件夹内的内容如图所示,v2.1核心代码还是mask-rcnn的东西,只添加了一个自定义数据集:
2.解压balloon_dataset.zip放置在v2.1文件夹内:
balloon_dataset.zip文件夹解压后还有一个__MACOSX文件夹,压缩包是苹果打包的,win解压后就多了一个文件夹,删除即可,然后v2.1文件夹内是这样式的: