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