leukemia / train_yolo.py
filmm's picture
Upload 4 files
14bb99e verified
raw
history blame contribute delete
459 Bytes
from ultralytics import YOLO
# โหลดโมเดล YOLOv8 เปล่า
model = YOLO("yolov8n.pt") # หรือ yolov8s.pt ถ้าเครื่องแรงขึ้น
# เทรนโมเดลใหม่ด้วย dataset ที่โหลดมา
model.train(
data="leukemia-detection/data.yaml", # path ไปยัง dataset.yaml
epochs=50,
imgsz=640,
batch=8,
name="leukemia-yolo-model",
)