med seg diff pytorch

MedSegDiff - بيتورتش

تنفيذ MedSegDiff في Pytorch - التقسيم الطبي SOTA خارج Baidu باستخدام DDPM والتكييف المحسن على مستوى الميزات، مع تصفية الميزات في مساحة فورييه.

تقدير

• StabilityAI على الرعاية السخية، وكذلك الرعاة الآخرين هناك

• Isamu وDaniel على إضافة نص تدريبي لمجموعة بيانات آفات الجلد!

ثَبَّتَ

$ pip install med-seg-diff-pytorch

الاستخدام

 import torch
from med_seg_diff_pytorch import Unet , MedSegDiff

model = Unet (
    dim = 64 ,
    image_size = 128 ,
    mask_channels = 1 ,          # segmentation has 1 channel
    input_img_channels = 3 ,     # input images have 3 channels
    dim_mults = ( 1 , 2 , 4 , 8 )
)

diffusion = MedSegDiff (
    model ,
    timesteps = 1000
). cuda ()

segmented_imgs = torch . rand ( 8 , 1 , 128 , 128 )  # inputs are normalized from 0 to 1
input_imgs = torch . rand ( 8 , 3 , 128 , 128 )

loss = diffusion ( segmented_imgs , input_imgs )
loss . backward ()

# after a lot of training

pred = diffusion . sample ( input_imgs )     # pass in your unsegmented images
pred . shape                              # predicted segmented images - (8, 3, 128, 128) 

تمرين

الأمر بالتشغيل

accelerate launch driver.py --mask_channels=1 --input_img_channels=3 --image_size=64 --data_path= ' ./data ' --dim=64 --epochs=100 --batch_size=1 --scale_lr --gradient_accumulation_steps=4

إذا كنت تريد إضافة حالة ذاتية حيث قمنا بتكييف القناع الموجود لدينا حتى الآن، فافعل --self_condition

ما يجب القيام به

• بعض أكواد التدريب الأساسية، مع حصول المدرب على مجموعة بيانات مخصصة مصممة خصيصًا لتنسيقات الصور الطبية - بفضل @isamu-isozaki
• محول كامل من أي عمق في المنتصف، كما هو الحال في الانتشار البسيط

الاستشهادات

 @article { Wu2022MedSegDiffMI ,
    title   = { MedSegDiff: Medical Image Segmentation with Diffusion Probabilistic Model } ,
    author  = { Junde Wu and Huihui Fang and Yu Zhang and Yehui Yang and Yanwu Xu } ,
    journal = { ArXiv } ,
    year    = { 2022 } ,
    volume  = { abs/2211.00611 }
}

 @inproceedings { Hoogeboom2023simpleDE ,
    title   = { simple diffusion: End-to-end diffusion for high resolution images } ,
    author  = { Emiel Hoogeboom and Jonathan Heek and Tim Salimans } ,
    year    = { 2023 }
}