点击这里下载here。这个代码是给验证部门用的,有详细的使用介绍。
对u版yolov5代码的修改。更加符合验证部门的使用。
看下readme内容。
nohup python val.py \
--weights yolov5n.pt \
--data coco128.yaml \
--img 640 \
--plots \
--batch-size 32\
--name 20220618 > my_val_result.log 2>&1 &
path: ../datasets/coco128 # 【改】改成用来测试数据的目录,下面包括images/xxx,labels/xxx结构。
train: images/train2017 # 你只用来做验证,不用关系这个
val: images/train2017 # images/xxx 下面都是图像,对应的labels/xxx下面的标签,你不用管这么怎么来的,直接像兵兵要就行。
test: # 你只用来做验证,不用关系这个
# Classes
nc: 80 # number of classes # 【改】类别数,我这里4个类别,所以这里直接写4就行。
names: ['person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus', 'train', 'truck', 'boat', 'traffic light',
'fire hydrant', 'stop sign', 'parking meter', 'bench', 'bird', 'cat', 'dog', 'horse', 'sheep', 'cow',
'elephant', 'bear', 'zebra', 'giraffe', 'backpack', 'umbrella', 'handbag', 'tie', 'suitcase', 'frisbee',
'skis', 'snowboard', 'sports ball', 'kite', 'baseball bat', 'baseball glove', 'skateboard', 'surfboard',
'tennis racket', 'bottle', 'wine glass', 'cup', 'fork', 'knife', 'spoon', 'bowl', 'banana', 'apple',
'sandwich', 'orange', 'broccoli', 'carrot', 'hot dog', 'pizza', 'donut', 'cake', 'chair', 'couch',
'potted plant', 'bed', 'dining table', 'toilet', 'tv', 'laptop', 'mouse', 'remote', 'keyboard', 'cell phone',
'microwave', 'oven', 'toaster', 'sink', 'refrigerator', 'book', 'clock', 'vase', 'scissors', 'teddy bear',
'hair drier', 'toothbrush'] # 【改】类别标签,像我的测试就4个类别,把这个替换掉就行了
# Download script/URL (optional)
download: https://ultralytics.com/assets/coco128.zip # 不用管,可以删掉也可以。
15服务器上
source /home/jian/.bashrc && conda activate lane
sh 02_val.sh
”runs/val/20220618“