32 lines
No EOL
523 B
Python
32 lines
No EOL
523 B
Python
from ultralytics import YOLO
|
|
|
|
# Load a model
|
|
model = YOLO("yolo26n.pt") # load a pretrained model (recommended for training)
|
|
EPOCHS = 10
|
|
IMG_SIZE = 640
|
|
BATCH_SIZE = 8
|
|
WORKERS = 2
|
|
PATIENCE = 5
|
|
SEED = 42
|
|
# Train the model
|
|
results = model.train(data="/Users/nguyenhongvan/recongnize/data/SKU/data_kaggle.yaml", epochs=EPOCHS,
|
|
imgsz=IMG_SIZE,
|
|
|
|
batch=BATCH_SIZE,
|
|
|
|
workers=WORKERS,
|
|
|
|
patience=PATIENCE,
|
|
|
|
cache=True,
|
|
|
|
amp=True,
|
|
|
|
cos_lr=True,
|
|
|
|
optimizer="AdamW",
|
|
|
|
seed=SEED,
|
|
|
|
exist_ok=True
|
|
) |