-
Notifications
You must be signed in to change notification settings - Fork 0
/
apart.py
79 lines (68 loc) · 3.9 KB
/
apart.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
import os
import SimpleITK as sitk
from shutil import copy
root="/jhcnas1/xinyi/zhongshan_380/after_370"
patients= os.listdir(root)
for patient in patients:
sequences=os.listdir(os.path.join(root,patient))
sequences.sort()
for sequence in sequences:
if 'DIXON' in sequence and 'Vane' not in sequence and 'in' not in sequence and '_W' not in sequence:
reader = sitk.ImageSeriesReader()
dicom_names = reader.GetGDCMSeriesFileNames(
os.path.join(root,patient,sequence))
if 'COR' or 'cor' in sequence :
if len(dicom_names)>50:
mid = len(dicom_names) // 2
sequence_front=sequence+'_W'
sequence_behind=sequence+'_in'
if not os.path.isdir(os.path.join(root,patient,sequence_front)):
os.makedirs(os.path.join(root,patient,sequence_front))
if not os.path.isdir(os.path.join(root,patient,sequence_behind)):
os.makedirs(os.path.join(root,patient,sequence_behind))
# 前半部分
for i in range(mid):
fname = dicom_names[i]
copy(os.path.join(root,patient, sequence, fname), os.path.join(root, patient,sequence_front,os.path.basename(fname)))
# 后半部分
for i in range(mid, len(dicom_names)):
fname = dicom_names[i]
copy(os.path.join(root,patient, sequence, fname), os.path.join(root, patient,sequence_behind,os.path.basename(fname)))
elif 'T2' in sequence:
if len(dicom_names)>50:
mid = len(dicom_names) // 2
sequence_front = sequence + '_W'
sequence_behind = sequence + '_in'
if not os.path.isdir(os.path.join(root, patient, sequence_front)):
os.makedirs(os.path.join(root, patient, sequence_front))
if not os.path.isdir(os.path.join(root, patient, sequence_behind)):
os.makedirs(os.path.join(root, patient, sequence_behind))
# 前半部分
for i in range(mid):
fname = dicom_names[i]
copy(os.path.join(root, patient, sequence, fname),
os.path.join(root, patient, sequence_front, os.path.basename(fname)))
# 后半部分
for i in range(mid, len(dicom_names)):
fname = dicom_names[i]
copy(os.path.join(root, patient, sequence, fname),
os.path.join(root, patient, sequence_behind, os.path.basename(fname)))
else:
if len(dicom_names)>100:
mid = len(dicom_names) // 2
sequence_front = sequence + '_W'
sequence_behind = sequence + '_in'
if not os.path.isdir(os.path.join(root, patient, sequence_front)):
os.makedirs(os.path.join(root, patient, sequence_front))
if not os.path.isdir(os.path.join(root, patient, sequence_behind)):
os.makedirs(os.path.join(root, patient, sequence_behind))
# 前半部分
for i in range(mid):
fname = dicom_names[i]
copy(os.path.join(root, patient, sequence, fname),
os.path.join(root, patient, sequence_front, os.path.basename(fname)))
# 后半部分
for i in range(mid, len(dicom_names)):
fname = dicom_names[i]
copy(os.path.join(root, patient, sequence, fname),
os.path.join(root, patient, sequence_behind, os.path.basename(fname)))