Idk
This commit is contained in:
18
rpi/training/training.py
Normal file
18
rpi/training/training.py
Normal file
@@ -0,0 +1,18 @@
|
||||
from ultralytics import YOLO
|
||||
|
||||
def main():
|
||||
model = YOLO("models/yolo11n.pt")
|
||||
model.train(
|
||||
data="./datasets/unified/data.yaml",
|
||||
epochs=200,
|
||||
patience=30,
|
||||
imgsz=640,
|
||||
batch=12,
|
||||
device=0,
|
||||
project="result",
|
||||
name="unified-training",
|
||||
exist_ok=True
|
||||
)
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user