U2net在DUTS数据集的显著性目标检测分割效果
本项目利用U2net网络模型对DUTS数据集中的显著性目标进行检测分割,并重新实现了预处理transform、计算混淆矩阵等方法,同时对预训练权重进行了迁移学习。通过对dataset的init方法进行重新编写,本项目能够实现其他的U2net网络中的二分类任务。详细性能数据可在本文中查看。
下载地址
用户评论