-
Notifications
You must be signed in to change notification settings - Fork 0
/
cilia_sizes.py
159 lines (125 loc) · 5.7 KB
/
cilia_sizes.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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
# Copyright 2018, Anthony Westbrook <anthony.westbrook@unh.edu>, University of New Hampshire
# To install this script as a FIJI macro, perform the following steps:
#
# 1. From the "Plugins" menu, select "Install..."
# 2. Select this script (cilia_sizes.py)
# 3. Accept the default installation directory
# 4. Restart FIJI
# 5. The plugin will be available to run from the "Plugins" menu at the bottom
from ij import IJ, WindowManager
from ij.measure import ResultsTable
from fiji.util.gui import GenericDialogPlus
from trainableSegmentation import WekaSegmentation
import os
HEADER = "File\tProbability\tUnit\tIndex\tArea\tPerimeter\tMajor\tMinor\tAngle\tRatio"
RESULTS_FILE = "results.csv"
DEFAULT_DIR = ""
DEFAULT_TRAIN = ""
def registerWindow(passTitle, passWindows):
passWindows.add(passTitle)
IJ.selectWindow(passTitle)
def closeWindows(passWindows):
for window in passWindows:
IJ.selectWindow(window)
IJ.run("Close")
def setupMeasurements():
options = "area perimeter fit"
IJ.run("Set Measurements...", options)
def optionsDialog():
dialog = GenericDialogPlus("Cilia Sizes")
dialog.addDirectoryField("Image Directory", DEFAULT_DIR)
dialog.addFileField("Training Model", DEFAULT_TRAIN)
dialog.addStringField("Output Subdirectory", "output", 20)
dialog.addStringField("Confocal Channel", "1", 20)
dialog.addStringField("Probability Threshold", "0.67", 20)
dialog.addStringField("Minimum Pixel Size", "2", 20)
dialog.showDialog()
# Check if canceled
if dialog.wasCanceled(): return None
textVals = [x.text for x in dialog.getStringFields()]
return textVals
def prepareImage(passDir, passFile):
# Attempt to open as image, exit if not
fullPath = os.path.join(passDir, passFile)
retImage = IJ.openImage(fullPath)
if not retImage: return None
retImage.show()
return retImage
def finalizeImage(passImage):
passImage.changes = False
passImage.close()
def analyzeImage(passImage, passModel, passChannel, passProbability, passPixels, passOutput):
retResults = list()
windows = set()
# Register current window
registerWindow(passImage.title, windows)
# Extract the requested channel
IJ.run("Z Project...", "projection=[Max Intensity]");
registerWindow("MAX_" + passImage.title, windows)
IJ.run("Duplicate...", "title=temp")
registerWindow("temp", windows)
# Apply WEKA training model to image
wekaSeg = WekaSegmentation(WindowManager.getCurrentImage())
wekaSeg.loadClassifier(passModel)
wekaSeg.applyClassifier(True)
# Extract first slice of probability map
wekaImg = wekaSeg.getClassifiedImage();
wekaImg.show()
registerWindow("Probability maps", windows)
IJ.setSlice(1)
IJ.run("Duplicate...", "title=temp2")
registerWindow("temp2", windows)
# Apply threshold and save
IJ.setThreshold(passProbability, 1, "Black & White")
fileParts = passImage.getTitle().split(".")
IJ.save(os.path.join(passOutput, "{0}-probmap.png".format(fileParts[0], '.'.join(fileParts[1:]))))
# Perform particle analysis and save
IJ.run("Analyze Particles...", "size={0}-Infinity show=Outlines pixel clear".format(passPixels))
registerWindow("Drawing of temp2", windows)
IJ.save(os.path.join(passOutput, "{0}-particles.png".format(fileParts[0], '.'.join(fileParts[1:]))))
# Get measurements
tableResults = ResultsTable.getResultsTable()
for rowIdx in range(tableResults.size()):
retResults.append(tableResults.getRowAsString(rowIdx).split())
retResults[-1].insert(0, WindowManager.getCurrentImage().getCalibration().unit)
retResults[-1].append(float(retResults[-1][4])/float(retResults[-1][3]))
# Close windows
closeWindows(windows)
return retResults
def processImages(passOptions):
optImageDir = passOptions[0]
optModel = passOptions[1]
optOutput = passOptions[2]
optChannel = int(passOptions[3])
optProbability = 1.0 - float(passOptions[4])
optPixels = int(passOptions[5])
retResults = dict()
# Iterate through all images in chosen directory
root, dirs, files = next(os.walk(optImageDir))
for curFile in files:
# Prepare image
curImage = prepareImage(root, curFile)
if not curImage: continue
# Analyze image
retResults[curFile] = analyzeImage(curImage, optModel, optChannel, optProbability, optPixels, os.path.join(options[0], optOutput))
return retResults
def writeResults(passResults, passOutput, passProbability):
# Create results TSV
with open(passOutput, "w") as fileHandle:
fileHandle.write("{0}\n".format(HEADER))
for image in passResults:
for row in passResults[image]:
fieldText = "\t".join(map(unicode, row))
fileHandle.write(u"{0}\t{1}\t{2}\n".format(image, passProbability, fieldText).encode("utf8"))
# Setup measurements to record in CSV
setupMeasurements()
# Present user definable options, then process
options = optionsDialog()
if options:
# Prepare output directory
if not os.path.exists(os.path.join(options[0], options[2])):
os.makedirs(os.path.join(options[0], options[2]))
# Analyze images
results = processImages(options)
writeResults(results, os.path.join(options[0], options[2], RESULTS_FILE), options[4])
IJ.showMessage("Analysis Complete!")