Spaces:
Sleeping
Sleeping
| 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") | |