PCB表面缺陷检测项目 | 轻量化PCB表面缺陷检测算法实现

• 项目应用场景
  • 面向 PCB 表面缺陷检测场景，包括漏孔 missing hole、咬伤 mouse bite、开路 open circuit、短路 short、支线 spur、杂铜 spurious copper 六种缺陷类型。

• 项目效果：

• 项目细节 ==> 具体参见项目 README.md

        (1) 下载模型

        (2) 安装依赖，包括 python、cuda、cudnn、opencv、tfplot、tensorflow 等，详见 requirements.txt

pip install -r requirements.txt
# pipenv run pip install -r requirements.txt

        (3) 编译

cd ${PWD}/libs/box_utils/cython_utils
python setup.py build_ext --inplace
# pipenv run python setup.py build_ext --inplace

        (4) 执行检测

mkdir -p ./input ./output
 
cd $PWD/tools
 
python inference.py --data_dir='../input' 
                    --save_dir='../output' 
                    --GPU='0'

        (5) 同时项目还执行训练自定义数据集

• 项目获取
  • ​​​​​​​https://download.csdn.net/download/weixin_42405819/89050012