From f1399395a90566ddcd72afc60cea2a26fbc04e49 Mon Sep 17 00:00:00 2001 From: HappyZ Date: Mon, 10 Jun 2019 11:58:20 -0500 Subject: [PATCH] bug fix --- ...tract_only_enough_data_with_groundtruth.py | 25 ++++++++++--------- 1 file changed, 13 insertions(+), 12 deletions(-) diff --git a/opencellid_parsing/003.extract_only_enough_data_with_groundtruth.py b/opencellid_parsing/003.extract_only_enough_data_with_groundtruth.py index 7cb41b7..e39b755 100644 --- a/opencellid_parsing/003.extract_only_enough_data_with_groundtruth.py +++ b/opencellid_parsing/003.extract_only_enough_data_with_groundtruth.py @@ -4,6 +4,19 @@ import os import sys import pickle +############################ +# extract measurements +############################ +folderpath = sys.argv[1] +pickle_fp_with_gt = "{}/all_with_groundtruth.pickle".format(folderpath) + +if not os.path.isfile(pickle_fp_with_gt): + print("run 002 script first") + sys.exit(2) + + +data = pickle.load(open(pickle_fp_with_gt, 'rb')) + outputfolder = folderpath.rstrip('/') + '_extracted' if not os.path.isdir(outputfolder): os.makedirs(outputfolder) @@ -16,18 +29,6 @@ headlines = { 'CDMA': headline + ',sid,nid,bid', } -############################ -# extract measurements -############################ -folderpath = sys.argv[1] -pickle_fp_with_gt = "{}/all_with_groundtruth.pickle".format(folderpath) - -if not os.path.isfile(pickle_fp_with_gt): - print("run 002 script first") - sys.exit(2) - -data = pickle.load(open(pickle_fp_with_gt, 'rb')) - print("extracting each file to disk (only groundtruth we cared)..") counter = 0 counter_valid = 0