摘要: 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 阅读全文
posted @ 2024-01-29 15:58 txwtech 阅读(139) 评论(1) 推荐(0) 编辑
摘要: 里面的文件含义如下: best.pt: 该文件是训练过程中最佳的模型权重参数文件,即validation loss最小的模型。在测试或者推理时,我们可以直接加载该文件来获取最优模型的参数。在weight文件夹中。 hyp.yaml: 该文件包含了模型的超参数,如学习率、权重衰减、数据增强等。我们可以 阅读全文
posted @ 2024-01-29 15:55 txwtech 阅读(169) 评论(0) 推荐(0) 编辑