220180T / model.py
NovaIZ's picture
added changes to app.py and model.py
774d622
from ultralytics import YOLO
from ultralytics import settings
from roboflow import Roboflow
rf = Roboflow(api_key="V6C1HgdOhbGH9M0SqFog")
project = rf.workspace("ata220180t").project("220180t_carlogos")
version = project.version(5)
dataset = version.download("yolov8")
settings.update({"wandb": True,
"comet": False,
"tensorboard": False}
)
model = YOLO("yolov8s.pt")
result = model.train(data="220180T_carlogos-5/data.yaml",
epochs=30,
save_period=1,
batch=50,
device= "cpu",
project='Car-Logos',
plots=True,
freeze=5,
lr0= 1e-7,
optimizer = "Adam" ,
cls = 0.1,
)
model = YOLO("car_logos.pt")
validation_results = model.val(data="220180T_carlogos.v5i.yolov8-obb/data.yaml", device="cpu")
model.save("car_logos.pt")