Refactor: Integrate backend API and normalize data
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>
This commit is contained in:
17
backend/test/Machine_Learning/train_deetction.py
Normal file
17
backend/test/Machine_Learning/train_deetction.py
Normal file
@@ -0,0 +1,17 @@
|
||||
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()
|
||||
Reference in New Issue
Block a user