Mercurial > repos > imgteam > labelimage2points
diff labelimage2points.py @ 0:39e6d6a84257 draft
planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tools/labelimage2points/ commit c3f4b766f03770f094fda6bda0a5882c0ebd4581
author | imgteam |
---|---|
date | Sat, 09 Feb 2019 14:36:17 -0500 |
parents | |
children | 07525a7d9ea0 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/labelimage2points.py Sat Feb 09 14:36:17 2019 -0500 @@ -0,0 +1,46 @@ +import argparse +import sys +import pandas as pd +import skimage.io +from skimage.measure import label +from skimage.data import checkerboard +import numpy as np + + + +def labelimage2points(input_file): + img_in = skimage.io.imread(input_file) + + #amount of regions + amount_label = np.max(img_in) + + # iterate over all regions in order to calc center of mass + center_mass = [] + for i in range(1,amount_label+1): + #get coordinates of region + coord = np.where(img_in==i) + # be carefull with x,y coordinates + center_mass.append([np.mean(coord[1]),np.mean(coord[0])]) + + #make data frame of detections + out_dataFrame = pd.DataFrame(center_mass) + + + #return + return(out_dataFrame) + + +if __name__ == "__main__": + parser = argparse.ArgumentParser() + parser.add_argument('input_file', help='input file') + parser.add_argument('out_file', help='out file (CSV)') + + args = parser.parse_args() + input_file = args.input_file + out_file = args.out_file + + #TOOL + out_dataFrame = labelimage2points(input_file) + + #Print to csv file + out_dataFrame.to_csv(out_file, index=False, header=False, sep="\t")