Pytorch
文章平均质量分 61
发量充裕の程序员
这个作者很懒,什么都没留下…
展开
-
【口罩识别三】关于train.py的一些解释
文章目录前言一、一些参数的解释总结前言主要用作毕设的学习过程记录一、一些参数的解释input_shape是指定输入到网络中的图片尺寸大小,可以在416,416到608,608两个值之间修改。mosaic值是否启用马赛克数据增强功能。Cosine_scheduler 余弦退火学习率。label_smoothing 标签平滑。anchors_path指的是yoloV4中总共九种大小的鲜艳框,每个网格上默认有三个鲜艳框。classes_path指向要分的类的标签文件原创 2021-04-07 16:04:28 · 505 阅读 · 0 评论 -
【口罩识别二】利用训练好的模型进行预测
文章目录前言一、更改模型路径和标签路径。二、进行预测1.调用predict.py文件预测总结前言主要用作毕设的学习过程记录一、更改模型路径和标签路径。修改在yolo.py文件下的model_path和classes_path。model_path指向新训练出来的模型pth文件,classes_path指向我们在前面模型训练时train.py中修改过的classes_path指向的那个文件。修改后的路径如下:二、进行预测1.调用predict.py文件预测运行结果如下:模型还在训练原创 2021-04-07 15:30:43 · 1128 阅读 · 0 评论