import warnings
warnings.filterwarnings('ignore')
from ultralytics import YOLO
if __name__ == '__main__':
model = YOLO('ultralytics/cfg/models/v8/yolov8n.yaml')
# model.load('yolov8n.pt') # loading pretrain weights
# # model = YOLO('yolov8n.yaml') # build a new model from YAML
# model = YOLO('runs/detect/train_v8n/exp_v8n-eyeball5/weights/last.pt') # load a pretrained model (recommended for training),配合resume=True,
# # model = YOLO('yolov8n.yaml').load('yolov8n.pt') # build from YAML and transfer weights
# model = all('ultralytics/cfg/models/v8/my-yolov8-CBAM-b1.yaml') # 从头开始构建新模型
model.train(data='./datasets/DYC/dataYAWDD.yaml',
# model.train(data='/home/hjj/Desktop/dataset/dataset_visdrone/data_exp.yaml',
# model.train(data='F/yolo/bilibilimodao/YOLOv8/ultralytics-20240627/ultralytics-main/data
YOLOv8-训练train.py参数设置
于 2024-06-29 15:43:54 首次发布