- 博客(3)
- 收藏
- 关注
原创 PaddleSeg模型的ONNX格式部署
近期在工作时遇到项目需要用ONNX格式实现PaddleSeg模型的本地部署,在此记录供日后自己可以参考。环境:Python3.8+RTX3060 + CUDA 11.4 + onnxruntime-GPU 1.11.0在贴码之前,PaddleSeg转ONNX模型的输出结果是一个三维 int64数组,分别对应N, H, W, 代表每一个像素点的分类,比如我的模型只有1个分类——缺陷,那么输出的数组中,0代表这个像素点是背景,1代表是缺陷。另外注意的是,如下图维度-1代表输入ONNX的尺寸可以是任意,.
2022-04-14 14:44:32 833
原创 PaddleSeg分割实战——汽车活塞卡簧检测项目
最近工作中遇到一个项目需要使用深度学习分割进行缺陷检测,一开始通过查百度使用了MRCNN算法,但运行过程中发现由于算法框架太陈旧,在搭建环境的时候会遇到各种千奇百怪的问题,遂放弃并转而投向百度Paddle的PaddleSeg分割算法。该笔记主要用于记录从搭建环境到运行结果的全过程和这其中遇到的一些坑,作为记录使用。1、背景及硬件条件:训练内容:汽车弹簧检测的附铜缺陷检测;训练环境:单卡GTX3060 12G+CUDA11.2+CUDNN8.X+Win10;训练算法:基于PaddlePaddl
2022-03-17 15:51:13 1328 3
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人