目标检测yolo算法
yolov5s:
从github官网下载yolov5的算法之后,配置好环境(pycharm安装包-CSDN博客),再下载权重文件,比如默认的yolov5s.pt;
运行当前文件(detect.py),就能看到exp文件夹中的结果
也可以自定义参数运行:
train.py和detect.py文件同时使用(训练自己的数据集)
先运行train.py(上面一行命令) 将其中的data.yaml(可自定义名字)中训练、验证的路径改成自己的数据库文件夹,权重文件看要求可大可小;
训练完之后生成best.pt和last.pt文件,
运行下面一行,导入检测图像和刚刚训练好的best.pt文件即可