Single Channel Breast Segmentation Unet
基於PyTorch的Unet圖像分割模型,支持多種編碼器架構,適用於語義分割任務
下載量 65
發布時間 : 10/15/2024
模型概述
這是一個基於PyTorch實現的Unet架構圖像分割模型,主要用於語義分割任務。模型支持多種預訓練編碼器,並提供了靈活的配置選項。
模型特點
多種編碼器支持
支持ResNet等主流編碼器架構,並可加載ImageNet預訓練權重
靈活的配置選項
可自定義編碼器深度、解碼器通道數、注意力機制等參數
批歸一化支持
解碼器支持批歸一化層,有助於訓練穩定性和模型性能
模型能力
圖像分割
語義分割
醫學圖像分析
使用案例
醫學影像
乳腺圖像分割
用於乳腺醫學圖像的分割任務
測試數據集IOU達到0.757
精選推薦AI模型
Llama 3 Typhoon V1.5x 8b Instruct
專為泰語設計的80億參數指令模型,性能媲美GPT-3.5-turbo,優化了應用場景、檢索增強生成、受限生成和推理任務
大型語言模型
Transformers 支持多種語言

L
scb10x
3,269
16
Cadet Tiny
Openrail
Cadet-Tiny是一個基於SODA數據集訓練的超小型對話模型,專為邊緣設備推理設計,體積僅為Cosmo-3B模型的2%左右。
對話系統
Transformers 英語

C
ToddGoldfarb
2,691
6
Roberta Base Chinese Extractive Qa
基於RoBERTa架構的中文抽取式問答模型,適用於從給定文本中提取答案的任務。
問答系統 中文
R
uer
2,694
98