Train, Validate and Detect using Weights & Biases Artifacts

[soumik12345]

Features Added

• Pass a model/checkpoint artifact (such as wandb/hand-detection-yolov5/run_2mza8br0_model:v0) to the weights parameter of train.py to load the weights from the artifact to the model for training
• Pass a model/checkpoint artifact (such as wandb/hand-detection-yolov5/run_2mza8br0_model:v0) to the weights parameter of val.py to load the weights from the artifact to the model for validation
• Pass a model/checkpoint artifact (such as wandb/hand-detection-yolov5/run_2mza8br0_model:v0) to the weights parameter of detect.py to load the weights from the artifact to the model for detection

[glenn-jocher]

@soumik12345 do you know how we can resolve the wandb>=0.12.2 issue below? https://github.com/ultralytics/yolov5/blob/2794483e091d50416289614a1a35f158fd25bee2/utils/loggers/init.py#L25-L29

Every training with W&B now displays a warning to the users now which is clearly not ideal:

[glenn-jocher]

@soumik12345 do you know how we can resolve the wandb>=0.12.2 issue below?

https://github.com/ultralytics/yolov5/blob/2794483e091d50416289614a1a35f158fd25bee2/utils/loggers/init.py#L25-L29

Every training with W&B now displays a warning to the users now which is clearly not ideal:

@soumik12345 have you seen my message above about wandb compatibility issues?

[soumik12345]

@soumik12345 do you know how we can resolve the wandb>=0.12.2 issue below? https://github.com/ultralytics/yolov5/blob/2794483e091d50416289614a1a35f158fd25bee2/utils/loggers/init.py#L25-L29

Every training with W&B now displays a warning to the users now which is clearly not ideal:

@soumik12345 have you seen my message above about wandb compatibility issues?

Hi @glenn-jocher, I think the version compatibility issue is caused because the newer version of the wandb client has some problems with nested configs (https://github.com/ultralytics/yolov5/blob/master/utils/loggers/wandb/wandb_utils.py#L191). Can you please tell me how to override the version constraint so I can test it out?

[glenn-jocher]

@soumik12345 thanks for looking into this! The version constraint is not enforced, i.e. there's no hard constraint, it's just that this warning message appears when wandb>0.12.10 is installed.