参考资料 ultralytics/utils/callbacks/dvc.py
备注
该文件可从https://github.com/ultralytics/ultralytics/blob/main/ ultralytics/utils/callbacks/dvc .py 获取。如果您发现问题,请通过提交 Pull Request🛠️ 帮助修复。谢谢🙏!
ultralytics.utils.callbacks.dvc._log_images(path, prefix='')
使用 DVCLive 记录指定路径下的图像,可选前缀。
源代码 ultralytics/utils/callbacks/dvc.py
ultralytics.utils.callbacks.dvc._log_plots(plots, prefix='')
如果之前未对图像进行处理,则记录绘图图像的训练进度。
源代码 ultralytics/utils/callbacks/dvc.py
ultralytics.utils.callbacks.dvc._log_confusion_matrix(validator)
使用 DVCLive 记录给定验证器的混淆矩阵。
源代码 ultralytics/utils/callbacks/dvc.py
ultralytics.utils.callbacks.dvc.on_pretrain_routine_start(trainer)
在训练前例行程序中为训练元数据初始化 DVCLive 记录器。
源代码 ultralytics/utils/callbacks/dvc.py
ultralytics.utils.callbacks.dvc.on_pretrain_routine_end(trainer)
ultralytics.utils.callbacks.dvc.on_train_start(trainer)
ultralytics.utils.callbacks.dvc.on_train_epoch_start(trainer)
ultralytics.utils.callbacks.dvc.on_fit_epoch_end(trainer)
记录训练指标和模型信息,并在每个拟合历元结束时进入下一步。
源代码 ultralytics/utils/callbacks/dvc.py
ultralytics.utils.callbacks.dvc.on_train_end(trainer)
如果 DVCLive 处于激活状态,则在训练结束时记录最佳指标、绘图和混淆矩阵。