本期给大家介绍的是124基于mask-rcnn的图像分割算法检测森林区域,效果图如下:
代码下载和视频演示地址;
124基于mask-rcnn的图像分割算法检测森林区域_哔哩哔哩_bilibili
代码整体是非常简便的,主要是data数据集文件夹和3个py文件。
运行01将json文件转标准格式文件.py能将图片数据集转成标准格式。
运行02train.py会对标准格式的数据集进行训练,训练好的模型保存在weights文件夹下。
运行03pyqt.py可以选择自己感兴趣的图片进行识别,通过调用上一步训练好的模型识别图片中森林区域。
注:
下载本代码环境自行安装
安装pyth