第1章:工欲善其事,必先利其器

各位同学,欢迎来到《叶片裂纹智能识别实战教程》。

我是你们的老朋友,一个在机械视觉领域摸爬滚打了十几年的工程师。今天咱们不聊虚的,直接上手干活。

做叶片裂纹识别,说白了就是让计算机学会“看”裂纹。但计算机不是人,它需要一套完整的“工具箱”才能工作。这一章,我就带你把这套工具箱备齐。

1.1 为什么是Anaconda?

很多新手上来就装Python,然后发现包管理乱成一锅粥。我个人习惯用Anaconda,为什么?

因为它自带包管理器conda,能帮你隔离不同项目的环境。你想想看,一个项目用PyTorch 1.0,另一个用2.0,要是装在一个环境里,迟早出问题。

我在项目中遇到过好几次,因为环境冲突导致模型训练到一半崩溃。从那以后,每个项目我都单独建一个环境,再也没出过这种幺蛾子。

核心要点:Anaconda = Python + 包管理 + 环境隔离。一步到位,省心省力。

1.2 安装Anaconda

去官网下载对应系统的安装包。Windows用户记得选64位版本。

安装时有个坑:一定要勾选“Add Anaconda to my PATH environment variable”。我曾经因为没勾这个,后面折腾了半天才找到原因。

注意:安装路径不要有中文或空格。否则后面装包时会报一些莫名其妙的错。

装完后,打开命令行(Windows用cmd或PowerShell,Mac/Linux用终端),输入:

conda --version

如果显示版本号,比如 conda 23.7.4,说明安装成功。

1.3 创建虚拟环境

接下来,我们为叶片裂纹识别项目单独建一个环境。我习惯用Python 3.8,兼容性好,坑少。

conda create -n blade_crack python=3.8

这里 blade_crack 是环境名,你可以随便起。激活环境:

conda activate blade_crack

看到命令行前面多了 (blade_crack) 字样,就说明环境激活成功了。

小技巧:每次打开新终端,记得先 conda activate blade_crack。不然你装的包都跑到base环境里去了。

1.4 安装OpenCV

OpenCV是图像处理的核心工具。读图、滤波、边缘检测,全靠它。

pip install opencv-python

验证安装:

python -c "import cv2; print(cv2.__version__)"

如果输出类似 4.8.1,说明装好了。

嗯,这里要注意:OpenCV有两个版本,opencv-python 是基础版,opencv-contrib-python 包含额外模块。我们做裂纹识别,基础版就够了。

1.5 安装PyTorch

PyTorch是深度学习框架,用来训练裂纹识别模型。安装前先确认你有没有NVIDIA显卡。

有显卡的同学,去 pytorch.org 选对应CUDA版本。比如CUDA 11.8:

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

没显卡的同学,装CPU版:

pip install torch torchvision torchaudio

个人经验:有显卡一定要装CUDA版。我刚开始用CPU训练,一张图片要跑3秒,换成GPU后只要0.1秒。效率差距不是一星半点。

验证安装:

python -c "import torch; print(torch.__version__)"

1.6 安装LabelImg

LabelImg是标注工具,用来框出图片中的裂纹位置。没有标注数据,模型就没法学。

pip install labelImg

启动:

labelImg

界面出来后,点“Open Dir”选择图片文件夹,“Change Save Dir”选择标注文件保存位置。标注格式选YOLO或VOC,我们后面会用到。

避坑指南:我曾经标注了200张图片,结果发现保存格式选错了,全部重来。所以开始前一定确认好格式。

1.7 GPU/CUDA环境验证

这是最关键的一步。很多同学装完CUDA,以为就能用了,结果模型还是在CPU上跑。

验证方法:

python -c "import torch; print(torch.cuda.is_available())"

如果输出 True,恭喜你,GPU环境配置成功。

如果输出 False,别慌。检查以下几点:

  • 显卡驱动是否安装?去NVIDIA官网下载对应驱动。
  • CUDA Toolkit是否安装?命令行输入 nvcc --version 查看。
  • PyTorch版本是否匹配CUDA版本?比如CUDA 11.8要装cu118版的PyTorch。

注意:不要用 conda install cudatoolkit 装CUDA,那只是运行时库。真正的CUDA Toolkit需要从NVIDIA官网下载。

1.8 知识体系总览

下面这张图,帮你理清本章的知识结构:

叶片裂纹智能识别环境 Anaconda 虚拟环境 OpenCV PyTorch LabelImg CUDA验证 环境配置流程 1. 安装Anaconda → 2. 创建虚拟环境 → 3. 安装OpenCV 4. 安装PyTorch → 5. 安装LabelImg → 6. 验证GPU/CUDA 每个步骤都有坑,跟着教程走,少走弯路

1.9 总结

这一章,我们完成了叶片裂纹识别项目的环境搭建。从Anaconda到虚拟环境,从OpenCV到PyTorch,再到LabelImg和CUDA验证,每一步我都踩过坑,也帮你们填了坑。

环境配好了,后面的事就好办了。下一章,我们开始真正接触叶片裂纹数据,看看这些图片长什么样。

记住:环境配置是基础中的基础。基础不牢,地动山摇。花点时间把这一步走稳,后面会省很多事。


公众号:蓝海资料掘金营,微信deep3321