本期给大家介绍的是151基于python目标检测的深海鱼fish检测。效果图如下:
代码下载和视频演示地址:
151基于python目标检测的深海鱼fish检测_哔哩哔哩_bilibili
代码是十分的简介,主要是3个py文件
运行01train.py能够将data文件夹下的数据集进行训练,训练好的模型保存在runs文件夹下。
运行02predict.py可以调用训练好的模型对单张图片进行检测。
运行03pyqt.py会有个pyqt的可视化界面,通过点击按钮可以加载感兴趣的图片进行识别。
注:
下载本代码环境自行安装
安装python和pytorch即可运行,
安装环境可参考博客: