This commit is contained in:
2025-12-21 12:06:28 +01:00
parent c0374b3890
commit 94b83fb018
32 changed files with 281 additions and 212 deletions

View File

@@ -0,0 +1,19 @@
from ultralytics import YOLO
def main():
model = YOLO("models/unified-nano.pt")
model.train(
data="./datasets/pieces/unified/data.yaml",
epochs=150,
patience=20,
imgsz=640,
batch=18,
save_period=10,
project="result",
name="unified-nano-refined",
exist_ok=True,
device = 0
)
if __name__ == "__main__":
main()