12 lines
414 B
Python
12 lines
414 B
Python
from ultralytics import YOLO
|
|
import time
|
|
start_time = time.time()
|
|
# model = YOLO("yolov8n.pt")
|
|
model = YOLO(r"C:\workspace\le-yolo\runs\detect\train40\weights\last.pt")
|
|
model.train(data="data.yaml", epochs=100, batch=8, device='cpu', imgsz=640, augment = True)
|
|
model.val()
|
|
print('训练完成')
|
|
end_time = time.time()
|
|
run_time = end_time - start_time
|
|
min = run_time // 60
|
|
print('运行时间：', min, '分钟') |