- 项目应用场景
- 面向 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