一种U-net模型的轻量化方法及装置
公开
摘要
本申请涉及深度学习技术领域，尤其涉及一种U‑net模型的轻量化方法及装置。该轻量化方法包括：获取U‑net模型；在所述U‑net模型中增加预裁剪因子层，得到第一模型；利用第一样本数据训练所述第一模型得到第二模型；对所述第二模型中预裁剪因子层的参数小于阈值的通道进行剪枝，得到第三模型。本申请实施例通过对U‑net模型进行剪枝，在加速网络模型推理的同时，保证了网络模型的准确率。
基本信息
专利标题 ：
一种U-net模型的轻量化方法及装置
专利标题（英）：
暂无
公开（公告）号 ：
CN114611674A
申请号 ：
CN202210283262.8
公开（公告）日 ：
2022-06-10
申请日 ：
2022-03-22
授权号 ：
暂无
授权日 ：
暂无
发明人 ：
邢志伟杨飞宇
申请人 ：
奥比中光科技集团股份有限公司
申请人地址 ：
广东省深圳市南山区粤海街道学府路63号高新区联合总部大厦11-13楼
代理机构 ：
深圳汉世知识产权代理事务所(特殊普通合伙)
代理人 ：
冷仔
优先权 ：
CN202210283262.8
主分类号 ：
G06N3/04
IPC分类号 ：
G06N3/04 G06N3/08 G06T7/136
IPC结构图谱
G
G部——物理
G06
计算；推算或计数
G06N
基于特定计算模型的计算机系统
G06N3/00
基于生物学模型的计算机系统
G06N3/02
采用神经网络模型
G06N3/04
体系结构，例如，互连拓扑
法律状态
2022-06-10 ：
公开
注：本法律状态信息仅供参考，即时准确的法律状态信息须到国家知识产权局办理专利登记簿副本。
文件下载
暂无PDF文件可下载
