From ed320e4c0d392def3b0db395865d10e38235be89 Mon Sep 17 00:00:00 2001 From: Ankle Date: Tue, 26 May 2026 12:54:41 +0800 Subject: [PATCH] Update README.md --- README.md | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/README.md b/README.md index 2a918d5..cc19425 100644 --- a/README.md +++ b/README.md @@ -1,3 +1,23 @@ +训练脚本/yolov5_birdproject/train.sh: + +export CUDA_VISIBLE_DEVICES=4,5,6,7 +python /yolov5_birdproject/train.py \ + --img 960 \ + --batch 16 \ + --epochs 100 \ + --data /yolov5_birdproject/data/coco128.yaml \ + --weight /yolov5_birdproject/yolov5s6.pt \ + --name KTY_yolov5s6_fromYNDC_addKTYneg_e100 \ + --device 4,5,6,7 + +训练前找好空闲的GPU +--data是数据集配置文件,我们一般把类别数nc设置为5,其中标签0为"鸟" +--weight是预训练权重,我们通常用yolov5s6.pt +--name是输出文件夹的路径 + + +以下为官方readme +