-
Notifications
You must be signed in to change notification settings - Fork 0
/
train.py
21 lines (17 loc) · 676 Bytes
/
train.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
# from ultralytics import YOLO
from ultralytics import YOLO
import wandb
import argparse
wandb.login(key="25f9a4d45af7f2e6899294620bffa4be42f15b47")
wandb.init(
project="Yolov8",
name="DroneRgbFit",
entity="biyotteu",
)
if __name__ == '__main__':
# parser = argparse.ArgumentParser()
# parser.add_argument('--path',type=str,default='')
# parser.add_argument('--model',type=str,default='yolov8x.pt')
# args = parser.parse_args()
model = YOLO("/home/ubuntu/dev/repos/YoloV8/Yolov8/DroneRGB/weights/best.pt")
model.train(data="/home/ubuntu/dev/repos/YoloV8/fit.yaml",project="Yolov8",name="DroneRgbFit",epochs=200, imgsz=640, batch=16)