best.pt含义

里面的文件含义如下:

  • best.pt: 该文件是训练过程中最佳的模型权重参数文件,即validation loss最小的模型。在测试或者推理时,我们可以直接加载该文件来获取最优模型的参数。在weight文件夹中。
  • hyp.yaml: 该文件包含了模型的超参数,如学习率、权重衰减、数据增强等。我们可以通过修改该文件中的超参数来改变模型的训练行为。
  • last.pt: 该文件是训练结束时的模型权重参数文件。如果没有指定--nosave参数,则每个epoch结束时,该文件都会被更新为最新的模型参数。在weight文件夹中。
  • opt.yaml: 该文件包含了模型训练时的各种选项和参数,如batch size、epoch数、优化器等。我们可以通过修改该文件中的参数来改变模型的训练行为。
  • results.csv: 该文件包含了模型在验证集上的表现结果,如mAP、precision、recall等。我们可以通过查看该文件来评估模型的性能。
  • train_batch{N}.jpg: 该文件夹包含了每个batch训练数据的可视化图像。在训练过程中,每个epoch会生成一次该文件夹,以便我们查看训练数据的情况。
  • weights: 该文件夹包含了训练过程中每个epoch的模型权重参数文件。在训练结束后,该文件夹中包含的所有文件都可以用于测试或者推理。
posted @ 2024-01-29 15:55  txwtech  阅读(169)  评论(0编辑  收藏  举报