46dea3304f
This commit integrates the backend API for fetching and updating report data. It also includes a normalization function to handle data consistency between the API and local storage. Co-authored-by: anthonymuncher <anthonymuncher@gmail.com>
18 lines
421 B
Python
18 lines
421 B
Python
from ultralytics import YOLO
|
|
|
|
def train():
|
|
model = YOLO("yolov12n.pt") # pretrained YOLOv8 small
|
|
model.train(
|
|
data="D:/CTF_Hackathon/gensprintai2025/pothole-detection-yolov12.v2i.yolov12/data.yaml",
|
|
epochs=10,
|
|
imgsz=512,
|
|
batch=8,
|
|
device=0,
|
|
project="checkpoints",
|
|
name="pothole_detector",
|
|
exist_ok=True
|
|
)
|
|
|
|
if __name__ == "__main__":
|
|
train()
|