Mercurial > repos > thomaswollmann > points2labelimage
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)