diff points2label.py @ 0:14525b8c8d67 draft default tip

planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/points2labelimage/ commit 356f2556bd629e5473d6711ec71372ea17d3e28f
author thomaswollmann
date Wed, 12 Dec 2018 05:12:06 -0500
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/points2label.py	Wed Dec 12 05:12:06 2018 -0500
@@ -0,0 +1,44 @@
+import argparse
+import sys
+import numpy as np
+import skimage.io
+import pandas as pd
+import warnings
+
+def points2label(labels, shape, output_file=None, has_header=False, is_TSV=False):
+    labelimg = np.zeros([shape[0], shape[1]], dtype=np.int32)
+
+    if is_TSV:
+        if has_header:
+            df = pd.read_csv(labels, sep='\t', skiprows=1, header=None)
+        else:
+            df = pd.read_csv(labels, sep='\t', header=None)
+    else:
+        if has_header:
+            df = pd.read_csv(labels, skiprows=1, header=None)
+        else:
+            df = pd.read_csv(labels, header=None)
+
+    for i in range(0, len(df)):
+        a_row = df.iloc[i]
+        labelimg[a_row[0], a_row[1]] = i+1
+
+    if output_file is not None:
+        with warnings.catch_warnings():
+            warnings.simplefilter("ignore")
+            skimage.io.imsave(output_file, labelimg, plugin='tifffile')
+    else:
+        return labelimg
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
+    parser.add_argument('label_file', type=argparse.FileType('r'), default=sys.stdin, help='label file')
+    parser.add_argument('out_file', type=argparse.FileType('w'), default=sys.stdin, help='out file')
+    parser.add_argument('org_file', type=argparse.FileType('r'), default=sys.stdin, help='input original file')
+    parser.add_argument('--has_header', dest='has_header', type=bool, default=False, help='label file has header')
+    parser.add_argument('--is_tsv', dest='is_tsv', type=bool, default=False, help='label file is TSV')
+    args = parser.parse_args()
+
+    original_shape = skimage.io.imread(args.org_file.name, plugin='tifffile').shape
+
+    points2label(args.label_file.name, original_shape, args.out_file.name, args.has_header, args.is_tsv)