Skip to content

Retrieving trained YOLOv5 weights from WandB #5968

@aniruddh10124

Description

@aniruddh10124

Search before asking

Question

After training the weights on YOLOv5 and saving the results on WandB, how do I save and retrieve those weights from WandB for using the trained weights to detect images in the test set? I know that some change needs to be done in the argument of !python detect.py --weight <argument>, but what should I write in place of <argument>?
Just to be clear, I am able to retrieve the trained weights from Google Drive, but my teams requires the weights to be stored and retrieved from WandB to streamline our work, which I am not able to do.
Thank you for your time.

Additional

No response

Metadata

Metadata

Assignees

No one assigned

    Labels

    questionFurther information is requested

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions