src.yolov5.utils.wandb_logging package

Submodules

src.yolov5.utils.wandb_logging.log_dataset module

src.yolov5.utils.wandb_logging.wandb_utils module

Utilities and tools for tracking runs with Weights & Biases.

class src.yolov5.utils.wandb_logging.wandb_utils.WandbLogger(opt, name, run_id, data_dict, job_type='Training')

Bases: object

Log training runs, datasets, models, and predictions to Weights & Biases.

This logger sends information to W&B at wandb.ai. By default, this information includes hyperparameters, system configuration and metrics, model metrics, and basic data metrics and analyses.

By providing additional command line arguments to train.py, datasets, models and predictions can also be logged.

For more on how this logger is used, see the Weights & Biases documentation: https://docs.wandb.com/guides/integrations/yolov5

check_and_upload_dataset(opt)

create_dataset_table(dataset, class_to_id, name='dataset')

download_dataset_artifact(path, alias)

download_model_artifact(opt)

end_epoch(best_result=False)

finish_run()

log(log_dict)

log_dataset_artifact(data_file, single_cls, project, overwrite_config=False)

log_model(path, opt, epoch, fitness_score, best_model=False)

log_training_progress(predn, path, names)

map_val_table_path()

setup_training(opt, data_dict)

src.yolov5.utils.wandb_logging.wandb_utils.check_wandb_config_file(data_config_file)

src.yolov5.utils.wandb_logging.wandb_utils.check_wandb_resume(opt)

src.yolov5.utils.wandb_logging.wandb_utils.get_run_info(run_path)

src.yolov5.utils.wandb_logging.wandb_utils.process_wandb_config_ddp_mode(opt)

src.yolov5.utils.wandb_logging.wandb_utils.remove_prefix(from_string, prefix='wandb-artifact://')

Module contents