training Namespace Reference

Functions
	copy_files (source_dir, dest_dir)

Variables
str	source_directory = "../models/"
str	destination_directory = "../models/old_models/"
	model = YOLO("../pretrained_yolo/yolov8n-seg.pt")
	results

Function Documentation

copy_files()

training.copy_files	(		source_dir,
			dest_dir
	)

Variable Documentation

destination_directory

str training.destination_directory = "../models/old_models/"

model

training.model = YOLO("../pretrained_yolo/yolov8n-seg.pt")

results

training.results

Initial value:

=  model.train(
    # data="/home/seame/ObjectDetectionAvoidance/dataset/data.yaml",
    data="/home/seame/ObjectDetectionAvoidance/prioridade/data.yaml",
    epochs=15,
    imgsz=320,
    hsv_h=0.015,        # hue
    hsv_s=0.3,        # saturation
    hsv_v=0.3,        # Brightness/contrast 
    translate=0.0,    # Disable translation
    scale=0.0,        # Disable scaling
    fliplr=0.0,       # Disable horizontal flip
    mosaic=0.0,       # Disable mosaic
    erasing=0.0,      # Disable random erasing
    # augment=transform,
    auto_augment=None,  # Disable auto-augmentation
    batch=16,
    device=0,
    workers=4,
    project="../models",
    name="test2",
    exist_ok=True,
    # freeze=10,  # Freeze backbone
    lr0=0.01,  
    patience=7,  # Early stopping
    weight_decay=0.0005
)

source_directory

str training.source_directory = "../models/"