首页
Python
Java
前端
数据库
Linux
Chatgpt专题
开发者工具箱
瞳仁专题
pytorch+u2net实现眼球的虹膜瞳仁分割
一、效果展示 (图片来源网络,如有侵权请联系删除) 二、技术摘要 用到的技术如下: 1、模型训练。使用pytorch版的u2net网络(https://gitee.com/throni0/U-2-Net)。 2、裁切眼睛图片,制作数据集。用到dlib68点模型,识别出眼部位置。后面u2net识别的时候,也需要先用dlib裁切,再送入模型。 3、图像操作。使用Pillow,简单方便。
阅读更多...