Spaces:
Running
Running
File size: 386 Bytes
2793310 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | import sys
from pathlib import Path
sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
from ultralytics import YOLO # noqa: E402
def main():
model = YOLO("ultralytics/cfg/models/master/master_wavec2f.yaml")
model.train(data="ultralytics/cfg/datasets/coco8.yaml", epochs=1, imgsz=640, batch=4, device="cpu")
model.val()
if __name__ == "__main__":
main()
|