nnUNet工业质检系统部署清单

环境准备

• [ ] 安装CUDA 11.4+与cuDNN 8.2+
• [ ] 创建conda环境：conda create -n nnunet_industrial python=3.9
• [ ] 安装依赖：pip install -r requirements.txt
• [ ] 配置环境变量：export nnUNet_raw_data_base=/data/datasets

模型训练

• [ ] 数据格式转换：使用nnunetv2/dataset_conversion/generate_dataset_json.py
• [ ] 运行预处理：nnUNetv2_plan_and_preprocess -d Dataset123_PhoneCase
• [ ] 启动训练：nnUNetv2_train Dataset123_PhoneCase 2d all --npz
• [ ] 模型导出：python -m nnunetv2.export_onnx -i results/Dataset123_PhoneCase

推理部署

• [ ] 转换TensorRT引擎：trtexec --onnx=model.onnx --saveEngine=model.trt --fp16
• [ ] 部署推理服务：python -m nnunetv2.inference.example_inference
• [ ] 连接PLC系统：配置nnunetv2/utilities/json_export.py输出格式

系统验证

• [ ] 运行基准测试：bash nnunetv2/tests/integration_tests/run_integration_test.sh
• [ ] 校准检测阈值：调整nnunetv2/postprocessing/remove_connected_components.py参数
• [ ] 连续运行测试：24小时稳定性验证