![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
目标检测小实践
文章平均质量分 86
rad~
然而事情没有这么简单
展开
-
目标检测之训练
前景概要:cuda与gpu的版本不匹配导致训练时模型出现错误。提示:本文主要记录了在模型训练中所产生的问题。模型训练以及在验证集上执行的代码如下所示,这是开启训练模型大门的钥匙。python train.py --data mask_data.yaml --cfg mask_yolov5s.yaml --weights pretrained/yolov5s.pt --epoch 50 --batch-size 10python val.py --data data/mask_data.yam原创 2022-04-16 15:24:53 · 848 阅读 · 0 评论 -
目标检测之数据处理
一 检查环境自己的电脑要安装该版本的torch(实际上该版本并不适合我的电脑)conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch 1.查看 CUDA 和 cuDNN 版本import torchprint(torch.version)print(torch.version.cuda)print(torch.backends.cudnn.version())不可以直接使用imp..原创 2022-04-16 15:20:10 · 446 阅读 · 0 评论