环境安装
-
Anaconda
-
可使用gpu,运行python脚本
-
-
pytorch
-
官网有安装命令行,已经做好了版本号匹配的。
-
-
cuba
-
在控制面板里查看版本号后去官网下载,向下兼容的,比如自己的cuda是12.2,用11.7也是ok的。这里建议不要下最高最新版,下载大家用的比较多的稳定版最好,出问题了教程也多。
-
-
cudnn
-
写可执行程序去推理onnx需要用到这个运行库,如果只是用python pytorh则用不到这个。
-
-
数据标注软件
-
labelimg
-
标注信息默认txt格式,这种格式可直接用于训练。
-
-
anylabeling
-
标注信息默认json格式,需要转换成txt,才能训练。
-
工具内可直接预测推导,比如在功能里选择yolov7模型,可帮助我们快速标记。
-
-
训练
- train.py 在Anaconda里执行才能用GPU资源。
- P/R值
- P/R值超低,甚至经常是0。
- 数据是否有误,检查runs/trains下生成的训练测试图jpg是否符合预期,检查标签与框框位置。
- 训练轮
- P/R值超低,甚至经常是0。