22 lines
1.0 KiB
Python
22 lines
1.0 KiB
Python
|
import warnings
|
|||
|
|
warnings.filterwarnings('ignore')
|
|||
|
|
from ultralytics import YOLO
|
|||
|
|
|
|||
|
|
# BILIBILI UP 魔傀面具
|
|||
|
|
# 验证参数官方详解链接：https://docs.ultralytics.com/modes/val/#usage-examples:~:text=of%20each%20category-,Arguments%20for%20YOLO%20Model%20Validation,-When%20validating%20YOLO
|
|||
|
|
|
|||
|
|
# 精度小数点保留位数修改问题可看<使用说明.md>下方的<YOLOV8源码常见疑问解答小课堂>第五点
|
|||
|
|
# 最终论文的参数量和计算量统一以这个脚本运行出来的为准
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
model = YOLO('runs/train/exp/weights/best.pt') # 选择训练好的权重路径
|
|||
|
|
model.val(data='/home/hjj/Desktop/dataset/dataset_visdrone/data.yaml',
|
|||
|
|
split='val', # split可以选择train、val、test 根据自己的数据集情况来选择.
|
|||
|
|
imgsz=640,
|
|||
|
|
batch=16,
|
|||
|
|
# iou=0.7,
|
|||
|
|
# rect=False,
|
|||
|
|
# save_json=True, # if you need to cal coco metrice
|
|||
|
|
project='runs/val',
|
|||
|
|
name='exp',
|
|||
|
|
)
|