S

Single Channel Breast Segmentation Unet

由AnikiFan開發
基於PyTorch的Unet圖像分割模型,支持多種編碼器架構,適用於語義分割任務
下載量 65
發布時間 : 10/15/2024

模型概述

這是一個基於PyTorch實現的Unet架構圖像分割模型,主要用於語義分割任務。模型支持多種預訓練編碼器,並提供了靈活的配置選項。

模型特點

多種編碼器支持
支持ResNet等主流編碼器架構,並可加載ImageNet預訓練權重
靈活的配置選項
可自定義編碼器深度、解碼器通道數、注意力機制等參數
批歸一化支持
解碼器支持批歸一化層,有助於訓練穩定性和模型性能

模型能力

圖像分割
語義分割
醫學圖像分析

使用案例

醫學影像
乳腺圖像分割
用於乳腺醫學圖像的分割任務
測試數據集IOU達到0.757
AIbase
智啟未來,您的人工智能解決方案智庫
© 2025AIbase