Initial commit for push folder

recognization/code/train.py

@@ -0,0 +1,32 @@
+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
+)