阿里云建站后台建站,天津设计网站公司,数据库做后台网站,asp网站开发实例pdf1.配置环境
1.
首先去官网下载yolov8的zip https://github.com/ultralytics/ultralytics 存放在我的目录下G:\bsh\yolov8 然后使用conda创建新的环境
conda create -n yolov8 python3.8
#然后激活环境
conda activate yolov8然后安装pytorch#xff0c;注意 #xff0c;py…1.配置环境
1.
首先去官网下载yolov8的zip https://github.com/ultralytics/ultralytics 存放在我的目录下G:\bsh\yolov8 然后使用conda创建新的环境
conda create -n yolov8 python3.8
#然后激活环境
conda activate yolov8然后安装pytorch注意 pytorch1.10.1以后的版本才支持rtdetr.pt的两个权重文件。 conda install pytorch1.10.1 torchvision0.11.2 torchaudio0.10.1 cudatoolkit10.2 -c pytorch
安装 ultralytics
pip install ultralytics然后我需要降低pillow的版本来适配我的pytorch因为报错ImportError: DLL load failed while importing _imaging: 找不到指定的模块。解决方法参考附录1如果你们不需要请略过
然后下载官网的权重文件 https://github.com/ultralytics/assets/releases 这里我下载了一堆进行测试 然后照一张图片进行测试 可以参考官方文档 https://docs.ultralytics.com/modes/predict/#inference-arguments 自己照一张图片放在G:\bsh\yolov8\datademo\bug.jpg里边
yolo detect predict modelweights/yolov8n.pt sourcedatademo\bus.jpg运行结果如下图 再拿一张自己随便拍的做测试只能说种类越来越多了之前yolov5那好像没有鼠标垫 ps 检测目录设置成文件夹就行。遍历文件夹里的所有文件进行检测。
yolo detect predict modelweights/yolov8n.pt sourcedatademo顺便附上yolov8支持的检测参数
Usage - sources:$ yolo modepredict modelyolov8n.pt source0 # webcamimg.jpg # imagevid.mp4 # videoscreen # screenshotpath/ # directorylist.txt # list of imageslist.streams # list of streamspath/*.jpg # globhttps://youtu.be/Zgi9g1ksQHc # YouTubertsp://example.com/media.mp4 # RTSP, RTMP, HTTP streamUsage - formats:$ yolo modepredict modelyolov8n.pt # PyTorchyolov8n.torchscript # TorchScriptyolov8n.onnx # ONNX Runtime or OpenCV DNN with dnnTrueyolov8n_openvino_model # OpenVINOyolov8n.engine # TensorRTyolov8n.mlpackage # CoreML (macOS-only)yolov8n_saved_model # TensorFlow SavedModelyolov8n.pb # TensorFlow GraphDefyolov8n.tflite # TensorFlow Liteyolov8n_edgetpu.tflite # TensorFlow Edge TPUyolov8n_paddle_model # PaddlePaddle测了刚才一圈这些模型除了yolov_nas的三个模型都可以正常使用据作者说暂时不打算加入yolo-NAS那不知道为什么会有推理的模型好奇怪
我还意外发现yolov8s可以检测到键盘其他的不行
2.训练自己的数据集
正在更新ing 2023年8月11日 还是使用yolov5准备的数据集准备方法可以看我之前的文章 http://t.csdn.cn/MhxVF 然后运行代码 yolo taskdetect modetrain modelweights/yolov8s.pt data./data/datazaoj1.yaml batch4 epochs50 imgsz640 workers4 device0 close_mosaic50这里要把batch和worker调小因为我的GTX 1060的显卡只有6G的显存不然 就会电脑死机记得训练的过程中不要打开谷歌浏览器这玩意占用内存太多了很容易导致训练失败。最后马赛克增强去掉不然显存也不够 训练结果比yolov5效果好多了
Validating runs\detect\train16\weights\best.pt...
Ultralytics YOLOv8.0.150 Python-3.8.17 torch-1.10.1 CUDA:0 (NVIDIA GeForce GTX 1060, 6144MiB)
Model summary (fused): 168 layers, 11127906 parameters, 0 gradientsClass Images Instances Box(P R mAP50 mAP50-95): 100%|██████████| 2/2 [00:0000:00, 7.51it/s]all 11 45 0.89 0.915 0.982 0.818yuanguojia 11 14 0.774 1 0.986 0.893naiguojia 11 6 0.866 1 0.995 0.879huanhuogai 11 7 0.848 1 0.995 0.795dianchi 11 6 0.883 1 0.995 0.829shuomingshu 11 6 0.966 1 0.995 0.851paomoban 11 6 1 0.491 0.924 0.662
Speed: 0.6ms preprocess, 14.5ms inference, 0.0ms loss, 1.5ms postprocess per image
Results saved to runs\detect\train16名称默认值描述batch16训练的批量大小modelnull训练模型权重可指定具体位置如yolov8n.ptyolov8n.yaml等epochs100训练的轮次imgsz640输入图像压缩后的尺寸devicenull用于训练的设备可选0或1或cpu等workers8多线程数据加载默认8datanull数据路径使用自定义的yaml文件或者官方yamllr0float初始学习率lrffloat最终学习率lr0 * lrfpatience50早期训练时准确率如果没有显著上升则停止的轮次saveTrue是否需要保存训练的模型和预测结果cacheFalse使用缓存进行数据加载可选True/ram, disk 或者 Falseprojectnull项目名称namenull实验的名称exist_okFalse是否覆盖现有实验pretrainedFalse是否使用预训练模型optimizer‘SGD’优化器可选[‘SGD’, ‘Adam’, ‘AdamW’, ‘RMSProp’]verboseFalse是否打印详细输出seed0重复性实验的随机种子deterministicTrue是否启用确定性模式single_clsFalse是否将多类数据训练为单类image_weightsFalse是否使用加权图像选择进行训练rectFalse是否支持矩形训练cos_lrFalse是否使用余弦学习率调度器close_mosaic10禁用最后 10 个 epoch 的马赛克增强resumeFalse是否从上一个检查点恢复训练lr00.01初始学习率(SGD1E-2, Adam1E-3)lrf0.01余弦退火超参数 (lr0 * lrf)momentum0.937学习率动量weight_decay0.0005权重衰减系数warmup_epochs3.0预热学习轮次warmup_momentum0.8预热学习率动量warmup_bias_lr0.1预热学习率box7.5giou损失的系数cls0.5分类损失的系数dfl1.5dfl损失的系数fl_gamma0.0焦点损失的gamma系数 (efficientDet默认gamma1.5)label_smoothing0.0标签平滑nbs64名义批次,比如实际批次为16,那么64/164,每4 次迭代才进行一次反向传播更新权重可以节约显存overlap_maskTrue训练期间掩码是否重叠仅限分割训练mask_ratio4掩码下采样率 (仅限分割训练)dropout0.0使用 dropout 正则化 (仅限分类训练
然后如果有验证机可以试一下效果
yolo taskdetect modeval modelruns/detect/train3/weights/best.pt datadata/fall.yaml device0
有这么多参数可以选择
名称默认值描述valTrue在训练期间验证/测试save_jsonFalse将结果保存到 JSON 文件save_hybridFalse保存标签的混合版本标签附加预测conf0.001用于检测的对象置信度阈值预测时默认 0.25 验证时默认0.001iou0.6NMS 的交并比 (IoU) 阈值max_det300每张图像的最大检测数halfTrue使用半精度 (FP16)dnnFalse使用 OpenCV DNN 进行 ONNX 推理plotsFalse在训练期间显示图片
其实如果正常训练完成会自动进行验证评估的就是最后显示的那一些精准度 然后推理测试模型预测
yolo taskdetect modepredict modelruns/detect/train3/weights/best.pt sourcedata/images device0
名称默认值描述source‘ultralytics/assets’图片或视频的源目录saveFalse是否保存结果showFalse是否显示结果save_txtFalse将结果保存为 .txt 文件save_confFalse保存带有置信度分数的结果save_cropFasle保存裁剪后的图像和结果conf0.3置信度阈值hide_labelsFalse隐藏标签hide_confFalse隐藏置信度分数vid_strideFalse视频帧率步幅line_thickness3边界框厚度像素visualizeFalse可视化模型特征augmentFalse将图像增强应用于预测源agnostic_nmsFalse类别不可知的 NMSretina_masksFalse使用高分辨率分割蒙版classesnull只显示某几类结果如class0, 或者 class[0,2,3]
结果比较满意 然后用rtdetr-x.pt试试进行对比 同样的训练代码worker4训练不起来改成了2也不行 然后就用rtdetr-l.pt测试了
yolo taskdetect modetrain modelweights/rtdetr-l.pt data./data/datazaoj1.yaml batch2 epochs50 imgsz640 workers1 device0 close_mosaic50这是我的参数 然后结果
Validating runs\detect\train20\weights\best.pt...
Ultralytics YOLOv8.0.150 Python-3.8.17 torch-1.10.1 CUDA:0 (NVIDIA GeForce GTX 1060, 6144MiB)
rt-detr-l summary: 498 layers, 31996070 parameters, 0 gradientsClass Images Instances Box(P R mAP50 mAP50-95): 100%|██████████| 3/3 [00:0000:00, 3.91it/s]all 11 45 0.687 0.941 0.951 0.738yuanguojia 11 14 0.699 1 0.885 0.79naiguojia 11 6 0.552 1 0.927 0.868huanhuogai 11 7 0.719 1 0.995 0.729dianchi 11 6 0.475 1 0.995 0.79shuomingshu 11 6 0.676 1 0.995 0.787paomoban 11 6 1 0.644 0.909 0.466
Speed: 1.0ms preprocess, 60.0ms inference, 0.0ms loss, 1.3ms postprocess per image
Results saved to runs\detect\train20就到这里然后去重新装个pytorch。 1.10.1 10.2cuda的torch官方配的是0.11.2的torchvision即使版本对的也是会出现这个问题。 后来给降到了1.10.0torch 0.11.0torchvision 正在配置ing
conda install pytorch1.10.0 torchvision0.11.0 torchaudio0.10.0 cudatoolkit10.2 -c pytorch附录1
1.pillow库报错
ImportError: DLL load failed while importing _imaging: 找不到指定的模块。
(yolov8) PS G:\bsh\yolov8 pip show pillow
Name: Pillow
Version: 9.3.0看出报错是因为pillow版本过高直接安装8.0版本即可
pip install pillow8.02.yolo_nas测试问题
安装pip install super_gradients之后还是报错LooseVersion distutils.version.LooseVersion AttributeError: module ‘distutils’ has no attribute ‘version’ 然后又安装的pip install setuptools59.5.0 文章转载自： http://www.morning.rnrfs.cn.gov.cn.rnrfs.cn http://www.morning.qzpkr.cn.gov.cn.qzpkr.cn http://www.morning.qrwnj.cn.gov.cn.qrwnj.cn http://www.morning.dgpxp.cn.gov.cn.dgpxp.cn http://www.morning.fbbpj.cn.gov.cn.fbbpj.cn http://www.morning.zbtfz.cn.gov.cn.zbtfz.cn http://www.morning.knpbr.cn.gov.cn.knpbr.cn http://www.morning.kmqwp.cn.gov.cn.kmqwp.cn http://www.morning.fxqjz.cn.gov.cn.fxqjz.cn http://www.morning.yfffg.cn.gov.cn.yfffg.cn http://www.morning.xjnw.cn.gov.cn.xjnw.cn http://www.morning.tnnfy.cn.gov.cn.tnnfy.cn http://www.morning.yqqxj26.cn.gov.cn.yqqxj26.cn http://www.morning.fgsqz.cn.gov.cn.fgsqz.cn http://www.morning.mhcys.cn.gov.cn.mhcys.cn http://www.morning.qfwzm.cn.gov.cn.qfwzm.cn http://www.morning.zxrtt.cn.gov.cn.zxrtt.cn http://www.morning.mhpkz.cn.gov.cn.mhpkz.cn http://www.morning.rmpkn.cn.gov.cn.rmpkn.cn http://www.morning.spbp.cn.gov.cn.spbp.cn http://www.morning.lkthj.cn.gov.cn.lkthj.cn http://www.morning.ggtkk.cn.gov.cn.ggtkk.cn http://www.morning.ybyln.cn.gov.cn.ybyln.cn http://www.morning.ptqbt.cn.gov.cn.ptqbt.cn http://www.morning.mzhhr.cn.gov.cn.mzhhr.cn http://www.morning.nbgfk.cn.gov.cn.nbgfk.cn http://www.morning.kpygy.cn.gov.cn.kpygy.cn http://www.morning.jwefry.cn.gov.cn.jwefry.cn http://www.morning.wwgpy.cn.gov.cn.wwgpy.cn http://www.morning.dkqyg.cn.gov.cn.dkqyg.cn http://www.morning.kjcll.cn.gov.cn.kjcll.cn http://www.morning.wslpk.cn.gov.cn.wslpk.cn http://www.morning.xlztn.cn.gov.cn.xlztn.cn http://www.morning.sqmbb.cn.gov.cn.sqmbb.cn http://www.morning.yjtnc.cn.gov.cn.yjtnc.cn http://www.morning.pxdgy.cn.gov.cn.pxdgy.cn http://www.morning.sqhtg.cn.gov.cn.sqhtg.cn http://www.morning.wtbzt.cn.gov.cn.wtbzt.cn http://www.morning.nzmqn.cn.gov.cn.nzmqn.cn http://www.morning.wdrxh.cn.gov.cn.wdrxh.cn http://www.morning.ylrxd.cn.gov.cn.ylrxd.cn http://www.morning.rwlnk.cn.gov.cn.rwlnk.cn http://www.morning.sqqds.cn.gov.cn.sqqds.cn http://www.morning.ykgkh.cn.gov.cn.ykgkh.cn http://www.morning.xxrgt.cn.gov.cn.xxrgt.cn http://www.morning.heleyo.com.gov.cn.heleyo.com http://www.morning.qzdxy.cn.gov.cn.qzdxy.cn http://www.morning.xmttd.cn.gov.cn.xmttd.cn http://www.morning.hhxpl.cn.gov.cn.hhxpl.cn http://www.morning.ztdlp.cn.gov.cn.ztdlp.cn http://www.morning.fyglg.cn.gov.cn.fyglg.cn http://www.morning.fbdkb.cn.gov.cn.fbdkb.cn http://www.morning.bgpb.cn.gov.cn.bgpb.cn http://www.morning.nysjb.cn.gov.cn.nysjb.cn http://www.morning.psxxp.cn.gov.cn.psxxp.cn http://www.morning.jbgzy.cn.gov.cn.jbgzy.cn http://www.morning.lgnrl.cn.gov.cn.lgnrl.cn http://www.morning.ykwgl.cn.gov.cn.ykwgl.cn http://www.morning.zfyr.cn.gov.cn.zfyr.cn http://www.morning.nxpqw.cn.gov.cn.nxpqw.cn http://www.morning.mlnbd.cn.gov.cn.mlnbd.cn http://www.morning.dfdhx.cn.gov.cn.dfdhx.cn http://www.morning.nlryq.cn.gov.cn.nlryq.cn http://www.morning.dmhs.cn.gov.cn.dmhs.cn http://www.morning.znkls.cn.gov.cn.znkls.cn http://www.morning.snmth.cn.gov.cn.snmth.cn http://www.morning.wzknt.cn.gov.cn.wzknt.cn http://www.morning.dxhdn.cn.gov.cn.dxhdn.cn http://www.morning.ghryk.cn.gov.cn.ghryk.cn http://www.morning.madamli.com.gov.cn.madamli.com http://www.morning.yrsg.cn.gov.cn.yrsg.cn http://www.morning.ltfnl.cn.gov.cn.ltfnl.cn http://www.morning.srmpc.cn.gov.cn.srmpc.cn http://www.morning.phtqr.cn.gov.cn.phtqr.cn http://www.morning.xlmgq.cn.gov.cn.xlmgq.cn http://www.morning.gyylt.cn.gov.cn.gyylt.cn http://www.morning.bqwsz.cn.gov.cn.bqwsz.cn http://www.morning.rmltt.cn.gov.cn.rmltt.cn http://www.morning.thpzn.cn.gov.cn.thpzn.cn http://www.morning.dpqwq.cn.gov.cn.dpqwq.cn
