annotate detection_viz.py @ 2:f46f64f9590d draft

planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit a47299b0f9de530b7501f72fae2dd43d4e7700ba
author imgteam
date Mon, 22 Jul 2019 11:22:03 -0400
parents ec1ca4d62192
children cb2d9de888a1
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
1 import argparse
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
2 import sys
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
3 import os
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
4 import csv
1
ec1ca4d62192 planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit b2acc1845a25828181597fe5b6982fe116a7796d
imgteam
parents: 0
diff changeset
5
0
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
6 import matplotlib
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
7 matplotlib.use('Agg')
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
8 import matplotlib.pyplot as plt
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
9
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
10 import skimage.io
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
11
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
12 def plot_circles(file_name, ax, color, stroke_size, radius):
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
13 resfile = open(file_name, 'rb')
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
14 rd = csv.reader(resfile, delimiter=',')
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
15 for row in rd:
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
16 circ = plt.Circle((int(row[1]), int(row[0])), lw=stroke_size, radius=radius, color=color, fill=False)
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
17 ax.add_patch(circ)
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
18 resfile.close()
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
19
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
20 def detection_viz(input_file, output_file, tp=None, fn=None, fp=None, stroke_size=3, circle_radius=50):
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
21 img = skimage.io.imread(input_file)
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
22
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
23 fig = plt.figure(figsize=(40, 40))
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
24 ax = fig.add_axes([0, 0, 1, 1])
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
25 ax.axis('off')
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
26
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
27 plt.imshow(img)
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
28 if tp is not None:
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
29 plot_circles(tp, ax, '#00FF00', stroke_size, circle_radius)
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
30 if fn is not None:
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
31 plot_circles(fn, ax, 'red', stroke_size, circle_radius)
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
32 if fp is not None:
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
33 plot_circles(fp, ax, 'darkorange', stroke_size, circle_radius)
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
34
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
35 fig.canvas.print_png(output_file, dpi=1800)
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
36
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
37 if __name__ == "__main__":
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
38 parser = argparse.ArgumentParser()
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
39 parser.add_argument('input_file', type=argparse.FileType('r'), help='original file')
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
40 # output file should not be of type argparse.FileType('w') sine it is created immediately in this case which leads to an error in renaming
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
41 parser.add_argument('out_file_str', type=str, help='string of output file name')
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
42 parser.add_argument('--tp', dest='input_tp_file', type=argparse.FileType('r'), help='input TP file')
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
43 parser.add_argument('--fn', dest='input_fn_file', type=argparse.FileType('r'), help='input FN file')
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
44 parser.add_argument('--fp', dest='input_fp_file', type=argparse.FileType('r'), help='input FP file')
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
45 parser.add_argument('--stroke_size', dest='thickness', default=3, type=float, help='stroke thickness')
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
46 parser.add_argument('--circle_radius', dest='circle_radius', type=float, default=50, help='circle radius')
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
47 args = parser.parse_args()
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
48
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
49 tp=None
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
50 if args.input_tp_file:
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
51 tp=args.input_tp_file.name
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
52 fn=None
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
53 if args.input_fn_file:
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
54 fn=args.input_fn_file.name
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
55 fp=None
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
56 if args.input_fp_file:
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
57 fp=args.input_fp_file.name
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
58
3f738d5d91bf planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/detection_viz/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
imgteam
parents:
diff changeset
59 detection_viz(args.input_file.name, args.out_file_str, tp=tp, fn=fn, fp=fp, stroke_size=args.thickness, circle_radius=args.circle_radius)