Skip to content

Question About Pre-trained Weights in Custom YOLOv8 Configuration #8

Open
@khandriod

Description

@khandriod

Hi,

I’m currently working with a custom YOLOv8 model configuration (yolov8_ResBlock_CBAM.yaml) and have a question regarding the use of pre-trained weights during training. After reviewing the YAML file, I noticed that there is no explicit mention of pre-trained weights. However, I did see that pretrained=True is set in the default.yaml file. If a pre-trained model is used, I am wondering which .pt file or model is being utilized.

My training script initializes the model using the following command:

model = YOLO(args.model) model.train(data=args.data_dir)

From the documentation and the provided YAML file, it's not entirely clear if pre-trained weights are being loaded by default or if the model is training from scratch with randomly initialized weights.

Could you clarify if, in the absence of an explicit weights specification in the YAML file, the YOLOv8 training process automatically uses pre-trained weights (e.g., COCO) by default? If so, is there a way to control this behavior explicitly in the configuration or training script?

Thanks for your great work.
Best regards

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions