search

rezazad68 / BCDU-Net

BCDU-Net : Medical Image Segmentation
712 stars 264 forks source link

There is a problem with the result of visual lung segmentation #39

Closed kingjames1155 closed 2 years ago

kingjames1155 commented 2 years ago

Dear author, thank you very much for opening up your code. I want to get the visualization results after lung segmentation. After I read the DCM file with simpleitk library, the results are shown in the figure below image Obviously, this result is wrong. model = M.BCDU_net_D3(input_size = (512,512,1)) model.load_weights('./weight/weight_lungD3.hdf5') vol = sitk.ReadImage('IM000090.dcm') volarray = sitk.GetArrayFromImage(vol) pre = model.predict(volarray)

This is the code I use. How do you visualize it, or what do you use as the input of the model?

rezazad68 commented 2 years ago

Please follow our preprocessing and postprocessing steps to generate the segmentation map for your data.