参考链接:
环境搭建
训练自己的数据集
完成上述两个步骤后进入yolo目录运行一下命令
# 进入yolo环境
conda activate yolov5-6
# 训练自己的数据集
python .\train.py --data=yolo_dataset/K.yaml
# 使用上面训练的数据集进行物体识别检测
python .\detect.py --data=yolo_dataset/K.yaml --source=yolo_dataset/images/train/4002.jpg --weights=runs/train/exp3/weights/best.pt
评论区