W

Water Meter Segmentation Unet Efficientnet B0 ReduceLROnPlateau 021224

由 nitidpong 开发
基于PyTorch的Unet图像分割模型,支持多种编码器架构
下载量 18
发布时间 : 12/3/2024
模型介绍
内容详情
替代品

模型简介

这是一个基于PyTorch实现的Unet图像分割模型,主要用于语义分割任务。模型支持多种预训练编码器,并提供了灵活的配置选项。

模型特点

多种编码器支持
支持EfficientNet等多种预训练编码器架构
灵活的配置选项
可自定义解码器通道数、是否使用批归一化等参数
预训练权重
支持使用ImageNet预训练权重初始化模型

模型能力

图像分割
语义分割
支持多种编码器架构

使用案例

工业检测
水表检测
用于水表图像的语义分割
测试集IoU达到0.824
医学影像
医学图像分割
可用于CT/MRI等医学图像的分割任务