happyz
/
propagation_gan
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

bug fix

This commit is contained in:
HappyZ 2019-06-10 11:58:20 -05:00
parent 5a91b25503
commit f1399395a9
1 changed files with 13 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
opencellid_parsing/003.extract_only_enough_data_with_groundtruth.py
View File

@ -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