o Ͱg @spddlZddlmZddlZddlZddlZddl Z ddZ ddZ ddd Z dd d Z ed kr6e d dSdS)NcCstjdddd\}}|d||dd|dd|dj|dd |dd |dd|d j|dd |d d |d d|d j|dd |d d|d d||S)a Creates a 4-panel matplotlib figure comparing the original image, ground truth mask, model output, and thresholded output. :param image: Original image (PIL Image or NumPy array) :param gt_mask: Ground truth mask (NumPy array) :param pred_mask: Model output (NumPy array) :param thresholded_mask: Thresholded prediction mask (NumPy array) :return: Matplotlib figure ) rfigsizerOriginal ImageoffgraycmapzGround Truth Maskz Model OutputzThresholded Output)pltsubplotsimshow set_titleaxis tight_layout)imagegt_mask pred_maskthresholded_maskfigaxesr/C:\muni\DP\seg1\Segment1\utils\visualization.pycreate_segmentation_plots rcCstjdddd\}}|d||dd|dd|dj|dd |dd |dd|d j|dd |d d |d d|S) aU Creates a 3-panel matplotlib figure comparing: 1. Original image 2. Predicted mask 3. Thresholded mask :param image: Original image (PIL Image or NumPy array) :param pred_mask: Model output mask (NumPy array) :param thresholded_mask: Thresholded prediction mask (NumPy array) :return: Matplotlib figure rr ) rrrrrr r zPredicted Maskr zThresholded Mask)rrrrr)rrrrrrrrcreate_segmentation_plot_test(s r#~/DP/data/syn_3d.py/extracted_data/1.nii.gzcCsttj|d|}ttj|d|}|}|}||||}||||}t }|j |ddd|j |dddt dS)Nrawsegr z NIfTI Image)colormapnameinfernoz NIfTI Mask) nibloadospathjoinZ get_fdataminmaxnapariViewer add_imagerun)rootfilenameimgmaskZdata_imgZ data_maskviewerrrrshow_niiDs r6cCs`d}t|d|d}t|d|d}t}|j|dd|j|ddtdS)Nz+C:/muni/DP/CS2Net/CS-Net/samples/validationz/image_z.npyz/pred_Image)r$Z Prediction)npr'r-r.r/r0)idxr)rpredr5rrr show_predsWs r;__main__r)rr )r)r(matplotlib.pyplotpyplotrZnibabelr&torchnumpyr8r-rrr6r;__name__rrrrs