selfuse

path = '/media/hp/4T/datasets/total001-300/PCAA/'
for category in ["HGG","LGG"]:
    categoryPath = os.path.join(path, category)
    for folder in os.listdir(categoryPath):

        label = 0 if category== "LGG" else 1
        featuresDict['id'] = folder
        featuresDict['label'] = label
        print(folder,label)
        folderPath = os.path.join(categoryPath,folder)
        segPath = os.path.join(folderPath,'pca_'+folder+'_'+'seg.nii')
        print(segPath)
        for mode in ['img']:
            originPath = os.path.join(folderPath,'pca_'+folder+'_'+mode+'.nii')
            print(originPath)

027 1
/media/hp/4T/datasets/total001-300/PCAA/HGG/027/pca_027_seg.nii
/media/hp/4T/datasets/total001-300/PCAA/HGG/027/pca_027_img.nii
063 1
/media/hp/4T/datasets/total001-300/PCAA/HGG/063/pca_063_seg.nii
/media/hp/4T/datasets/total001-300/PCAA/HGG/063/pca_063_img.nii
034 1
/media/hp/4T/datasets/total001-300/PCAA/HGG/034/pca_034_seg.nii
/media/hp/4T/datasets/total001-300/PCAA/HGG/034/pca_034_img.nii
044 0
/media/hp/4T/datasets/total001-300/PCAA/LGG/044/pca_044_seg.nii
/media/hp/4T/datasets/total001-300/PCAA/LGG/044/pca_044_img.nii
132 0
/media/hp/4T/datasets/total001-300/PCAA/LGG/132/pca_132_seg.nii
/media/hp/4T/datasets/total001-300/PCAA/LGG/132/pca_132_img.nii
105 0

```bash
import numpy as np
import pandas as pd
import matplotlib as mpl
import matplotlib.pyplot as plt
import seaborn as sns
df = pd.read_csv('./features/Features_Resnet.csv')

# Draw Plot
i=2
for i in range(500):
    

    plt.figure(figsize=(13,10), dpi= 80)
    sns.violinplot(x='label', y=df.columns[i], data=df, scale='width', inner='quartile')

    # Decoration
    plt.title(str(i), fontsize=22)
    plt.show()

def merge_nii(folder):
    arrs = []
    for file in sorted(os.listdir(root_dir)):
        if "img" not in file:
            image = sitk.ReadImage(os.path.join(root_dir, file))
            arrs.append(sitk.GetArrayFromImage(image))

    new_arr = np.zeros(arrs[0].shape, dtype=np.uint8)
    for i in range(len(arrs)):
        new_arr[arrs[i] > 0] = 1
    
    new_image = sitk.GetImageFromArray(new_arr)
    new_image.SetSpacing(image.GetSpacing())
    new_image.SetOrigin(image.GetOrigin())
    new_image.SetDirection(image.GetDirection())
    
    sitk.WriteImage(new_image, os.path.join(root_dir, "merged.nii"))

plt.figure(figsize=(16,16))
for i in range(1,6,2):
    IMG = train_dataset[50]
    img = IMG['img']
    label = IMG['label']
    
    img = np.transpose(img, (1,2,0))
    img = img*0.5 + 0.5
    plt.subplot(3,2,i),plt.imshow(img,'gray'),plt.title('img'),plt.xticks([]),plt.yticks([])
    plt.subplot(3,2,i+1),plt.imshow(label,'gray'),plt.title('label'),plt.xticks([]),plt.yticks([])
    plt.show