摘要:
PS D:\deep_learning\yolov5-master> D:\anaconda3\envs\pytorch\python.exe train.py --cfg qiqiu.yaml --epoch 10 --batch-size 2 --device cputrain: weights 阅读全文
摘要:
里面的文件含义如下: best.pt: 该文件是训练过程中最佳的模型权重参数文件,即validation loss最小的模型。在测试或者推理时,我们可以直接加载该文件来获取最优模型的参数。在weight文件夹中。 hyp.yaml: 该文件包含了模型的超参数,如学习率、权重衰减、数据增强等。我们可以 阅读全文