习惯了YOLOv5使用Python文件操作,用不惯YOLOv8删除了Python文件操作。
手搓一下
from ultralytics import YOLO
if __name__ == '__main__':
#设置运行模式
mode = "deploy"
if mode == "train":
#项目命名
projectname='YOLOv8-small'
# 加载一个模型
model = YOLO('yolov8s-P2.yaml') # 从YAML建立一个新模型
model = YOLO('yolov8s.pt') # 加载预训练模型(推荐用于训练)
# 训练模型
model.train(data="F://datasets//rebar//rebar.yaml", epochs=10, imgsz=640, name='F://Llandudno//train//'+projectname) # train the model
metrics = model.val(name='F://Llandudno//val//'+projectname) # evaluate model performance on the validation set
#results = model("https://ultralytics.com/images/bus.jpg") # 使用刚训练好图片进行预测
#path = model.export(format="onnx") # 刚训练好的模型导出格式
elif mode == "val":
# 项目命名
projectname = 'YOLOv8s'
# 加载模型
#model = YOLO('yolov8.pt') # 加载官方模型
model = YOLO('E://PycharmProjects//ultralytics-main//runs//detect//coc