You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
18 lines
576 B
18 lines
576 B
6 months ago
|
from ultralytics import YOLO
|
||
|
|
|
||
|
|
# 1. 모델 로드 (YOLOv8m 사용)
|
||
|
|
# model = YOLO('yolov8m.pt')
|
||
|
|
model = YOLO("/home/cuuva/git/Detection_Experiment/fashion_yolo/fashionpedia_exp/yolov8m_fashion_final/weights/best_fashion_16class.pt")
|
||
|
|
|
||
|
|
train_results = model.train(
|
||
|
|
data="/home/cuuva/git/Detection_Experiment/fashion_yolo/finetuning_exp/finetuning_dataset.yaml",
|
||
|
|
epochs=200,
|
||
|
|
imgsz=640,
|
||
|
|
batch=-1,
|
||
|
|
device="cuda",
|
||
|
|
optimizer='AdamW',
|
||
|
|
lr0=0.001,
|
||
|
|
patience=50,
|
||
|
|
project='finetuned_exp',
|
||
|
|
name='yolov8m_finetuned_16class',
|
||
|
|
)
|