Showing
6 changed files
with
147 additions
and
0 deletions
code/getTargets/getclass by filename.py
0 → 100644
| 1 | +import pandas as pd | ||
| 2 | +import os | ||
| 3 | +from natsort import natsorted | ||
| 4 | + | ||
| 5 | +csv = 'ce_train_targets.csv' | ||
| 6 | +data_path = 'ce_train' | ||
| 7 | + | ||
| 8 | +df = pd.read_csv(csv) | ||
| 9 | + | ||
| 10 | +#idx = df.index[df.iloc[:,0]=='BraTS19_CBICA_BHB_1_seg_flair_8.png'].tolist() | ||
| 11 | + | ||
| 12 | +# df = df.loc[df.iloc[:,0]=='BraTS19_CBICA_BHB_1_seg_flair_8.png'] | ||
| 13 | +# print(df.iloc[0, 1]) | ||
| 14 | + | ||
| 15 | + | ||
| 16 | +imgs = natsorted(os.listdir(data_path)) # img file list | ||
| 17 | +targets = [] | ||
| 18 | + | ||
| 19 | +for fname in imgs: | ||
| 20 | + row = df.loc[df['filename'] == fname] | ||
| 21 | + targets.append(row.iloc[0, 1]) | ||
| 22 | + | ||
| 23 | +#print(targets, len(targets)) | ||
| 24 | + | ||
| 25 | +#BraTS19_2013_10_1_seg_flair_0.png -> class 0 | ||
| 26 | +#BraTS19_CBICA_AYU_1_seg_flair_2.png -> class 1 | ||
| ... | \ No newline at end of file | ... | \ No newline at end of file |
code/getTargets/getclass by filename.txt
0 → 100644
File mode changed
code/getTargets/shape_exception_handling.m
0 → 100644
| 1 | +data_path = '..\data\MICCAI_BraTS_2019_Data_Training\HGG_seg_flair\BraTS19_CBICA_AVG_1_seg_flair.nii'; | ||
| 2 | +data = niftiread(data_path); | ||
| 3 | + | ||
| 4 | +cp_data = flipud(rot90(mat2gray(double(data(:,:,82))))); | ||
| 5 | + | ||
| 6 | +type = '.png'; | ||
| 7 | +filename = strcat('BraTS19_CBICA_AVG_1_seg_flair_8', type); % BraTS19_2013_2_1_seg_flair_c.png | ||
| 8 | +outpath = strcat('..\data\MICCAI_BraTS_2019_Data_Training\zero to valid\', filename); | ||
| 9 | +imwrite(cp_data, outpath); | ||
| 10 | + | ||
| 11 | + | ||
| 12 | +% st, end 10 | ||
| 13 | + | ||
| 14 | +%% (circle 0, ellipse 0) | ||
| 15 | +% BraTS19_CBICA_AOP_1_seg_flair_0 | ||
| 16 | +% BraTS19_CBICA_ASF_1_seg_flair_9 -> 8 | ||
| 17 | +% BraTS19_CBICA_ASR_1_seg_flair_8 | ||
| 18 | +% BraTS19_CBICA_ASR_1_seg_flair_9 | ||
| 19 | +% BraTS19_CBICA_AVG_1_seg_flair_9 -> 8 | ||
| 20 | +% | ||
| 21 | +% | ||
| 22 | +% | ||
| 23 | +% | ||
| 24 | +% | ||
| 25 | +% | ||
| 26 | +% | ||
| 27 | +% | ||
| 28 | +% | ||
| 29 | +% | ||
| 30 | +% | ||
| 31 | +% | ||
| ... | \ No newline at end of file | ... | \ No newline at end of file |
| 1 | +getname_path = '..\data\MICCAI_BraTS_2019_Data_Training\zero2\'; | ||
| 2 | +inputheader = '..\data\MICCAI_BraTS_2019_Data_Training\HGG_seg_flair\'; | ||
| 3 | +outfolder = '..\data\MICCAI_BraTS_2019_Data_Training\zero to valid2\'; | ||
| 4 | + | ||
| 5 | +files = dir(getname_path); | ||
| 6 | +id = {files.name}; | ||
| 7 | +% files + dir file | ||
| 8 | +flag = ~strcmp(id, '.') & ~strcmp(id, '..'); | ||
| 9 | +files = files(flag); | ||
| 10 | + | ||
| 11 | +% get filenames in getname_path folder | ||
| 12 | +for i = 1 : length(files) | ||
| 13 | + %id = split(files(i).name, '.nii'); | ||
| 14 | + fname = files(i).name; | ||
| 15 | + id = reverse(extractBetween(reverse(fname), 7, strlength(fname))); | ||
| 16 | + | ||
| 17 | + data_path = convertCharsToStrings(strcat(inputheader,'\', id, '.nii')); | ||
| 18 | + data = niftiread(data_path); | ||
| 19 | + | ||
| 20 | + fprintf('ID #%d = %s\n', i, data_path); | ||
| 21 | + | ||
| 22 | + idx = 1; | ||
| 23 | + frames = zeros(1,1); | ||
| 24 | + for j = 1 : z | ||
| 25 | + n_nonblack = numel(find(data(:,:,j) > 0)); | ||
| 26 | + if(n_nonblack > 70) | ||
| 27 | + frames(idx,1) = j; | ||
| 28 | + idx = idx + 1; | ||
| 29 | + end | ||
| 30 | + end | ||
| 31 | + | ||
| 32 | +% for j = z : -1 : 1 | ||
| 33 | +% n_nonblack = numel(find(data(:,:,j) > 0)); | ||
| 34 | +% if(n_nonblack > 50) | ||
| 35 | +% fprintf('%s: %d\n ','number: ', n_nonblack); | ||
| 36 | +% en = j; | ||
| 37 | +% break; | ||
| 38 | +% end | ||
| 39 | +% end | ||
| 40 | + | ||
| 41 | + | ||
| 42 | + c = 0; | ||
| 43 | + [nrow, ncol] = size(frames); | ||
| 44 | + step = round(nrow/11); | ||
| 45 | +% fprintf('%s: %d \n', 'nrow', nrow); | ||
| 46 | +% fprintf('%s: %d\n ','st: ', frames(1, 1)); | ||
| 47 | +% fprintf('%s: %d\n ','end: ', frames(nrow, 1)); | ||
| 48 | + %disp(frames); | ||
| 49 | + | ||
| 50 | + for k = 1 : step : step*10 | ||
| 51 | + %fprintf('%d ', k); | ||
| 52 | + fprintf('%d ', frames(k, 1)); | ||
| 53 | + type = '.png'; | ||
| 54 | + filename = strcat(id, '_', int2str(c), type); % BraTS19_2013_2_1_seg_flair_c.png | ||
| 55 | + outpath = convertCharsToStrings(strcat(outfolder, filename)); | ||
| 56 | + % typecase int16 to double, range[0, 1], rotate 90 and filp updown | ||
| 57 | + % range [0, 1] | ||
| 58 | + cp_data = flipud(rot90(mat2gray(double(data(:,:,frames(k, 1)))))); | ||
| 59 | +% M = max(cp_data(:)); | ||
| 60 | +% disp(M); | ||
| 61 | + imwrite(cp_data, outpath); | ||
| 62 | + | ||
| 63 | + c = c+ 1; | ||
| 64 | + end | ||
| 65 | + | ||
| 66 | + | ||
| 67 | +end | ||
| 68 | + |
code/getTargets/target per shape_csv.py
0 → 100644
| 1 | +import pandas as pd | ||
| 2 | +import os | ||
| 3 | +import csv | ||
| 4 | + | ||
| 5 | +class0_path = "../../data/MICCAI_BraTS_2019_Data_Training/circle_val" | ||
| 6 | +class1_path = "../../data/MICCAI_BraTS_2019_Data_Training/ellipse_val" | ||
| 7 | +target_csv = "../../data/MICCAI_BraTS_2019_Data_Training/ce_valid_targets.csv" | ||
| 8 | + | ||
| 9 | +f=open(target_csv, 'w', newline='') | ||
| 10 | +w=csv.writer(f) | ||
| 11 | + | ||
| 12 | +for path, dirs, files in os.walk(class0_path): | ||
| 13 | + for filename in files: | ||
| 14 | + print(filename) | ||
| 15 | + n_class = 0 | ||
| 16 | + w.writerow([filename, n_class]) | ||
| 17 | + | ||
| 18 | +for path, dirs, files in os.walk(class1_path): | ||
| 19 | + for filename in files: | ||
| 20 | + print(filename) | ||
| 21 | + n_class = 1 | ||
| 22 | + w.writerow([filename, n_class]) | ||
| ... | \ No newline at end of file | ... | \ No newline at end of file |
-
Please register or login to post a comment