Demo运行
环境配好后,把https://github.com/leekunhee/Mask_RCNN代码项目git克隆到本地(直接打包下载也行),在IDE打开Mask_RCNN文件夹
numpy在1.20版本之后就弃用np.bool了,我们安装的版本为numpy1.23.4,所以要把代码中的np.bool改成np.bool_
找到mrcnn
文件夹下的model.py
,把第1274行、1329行的np.bool
改成np.bool_
找到mrcnn
文件夹下的utils.py
,把第533、551、571、574行的np.bool
改成np.bool_
同理coco.py
、ncleus.py
、train_shapes.ipynb
里的np.bool
也要改,可以用IDE的查找功能但是一定不要一键全部替换,自己手动点进查找结果确认后面没有下划线的再替换
找到samples
目录下的demo.ipyb
打开运行,以为这次一切顺利了,环境版本都对应了,并且每一步的运行结果也都一致,但是到了最后一个cell的程序就卡住了。。。等了十分钟都没有输出,一看性能情况发现GPU沾满但是利用率为0。
No instances to display问题解决
接着上一步然后从头运行了,结果出现了新的问题,运行没有报错,但是输出*** No instances to display ***
解决方案参考 No instances to display ,修改步骤如下
修改mrcnn文件夹下的model.py,行数可能不一定跟我的一模一样自己对照前后的代码一样的位置,xx修改mrcnn文件夹下的utils.py,注释掉原来的引用并添加下面这行,然后修改下面两行划红线的位置
修改mrcnn文件夹下的visualize.py,图中划线位置
修改samples文件夹下的coco文件里的coco.py
每一个文件修改完后最好要手动按下保存,以防修改的地方没有自动保存下来
推理爆内存并且GPU利用率为0——待解决
按照以上步骤都修改完了然后继续从头运行,但是显存这个问题不知道怎么回事还是存在,而且推理结果几分钟才出来明显不合理。。。还是不知道问题出在哪里了,等我找到解决方案再来更新,或者有大佬知道问题所在的评论区指点一下,谢谢了!