| import warnings | |
| warnings.filterwarnings('ignore') | |
| from load_model import load_model | |
| def train(base_model,data_path,cfg_path): | |
| model = load_model(base_model)#这里直接加载yolo模型训练 | |
| model.train(data=data_path,#配置数据集的配置文件 | |
| cfg=cfg_path#数据增强、优化器、周期、模型保存、超参数等配置文件 | |
| ) | |
| if __name__ == '__main__': | |
| train(r"yolo11n.pt", | |
| r'..\yolo_hanzi_dataset\dataset.yaml', | |
| r'ultralytics\cfg\default.yaml') | |