发明公开
- 专利标题: 一种水电工程高陡边坡危岩落石数据集扩充方法
-
申请号: CN202311189817.3申请日: 2023-09-14
-
公开(公告)号: CN117197744A公开(公告)日: 2023-12-08
- 发明人: 王金国 , 王如宾 , 张贵科 , 徐卫亚 , 邓韶辉 , 张海龙 , 闫龙 , 王耘梓
- 申请人: 雅砻江流域水电开发有限公司 , 河海大学
- 申请人地址: 四川省成都市成华区双林路288号
- 专利权人: 雅砻江流域水电开发有限公司,河海大学
- 当前专利权人: 雅砻江流域水电开发有限公司,河海大学
- 当前专利权人地址: 四川省成都市成华区双林路288号
- 代理机构: 南京苏高专利商标事务所
- 代理商 郭楠
- 主分类号: G06V20/52
- IPC分类号: G06V20/52 ; G06V10/10 ; G06V10/764 ; G06V10/82 ; G06V10/26 ; G06V10/80 ; G06N3/0475 ; G06N3/094
摘要:
本发明公开了一种水电工程高陡边坡危岩落石数据集扩充方法,该方法首先采集高陡边坡危岩落石图像,利用全卷积神经网络FCN对危岩落石图像进行像素级的分类及处理,形成高陡边坡危岩落石图像真实数据集;通过融合扩散模型DDPM与生成对抗神经网络GAN,训练随机生成大量的高陡边坡危岩落石图像扩充数据集,并导入YOLOV7目标检测算法,达到实现危岩落石的精准识别与监测预警的目的。本发明通过融合扩散模型DDPM与生成对抗神经网络GAN扩充危岩落石目标检测数据集,融合扩散模型DDPM帮助GAN克服训练不稳定的问题,本发明解决了目前高陡边坡危岩落石目标检测数据集难扩充的技术问题。