def load_Bag_Folders(bag_folders_path_meta_path, bag_folders_rgb_1to4_path):
BF_dic = {}
BF_dic_keys_weights = []
bag_folders_paths_list = sorted(gg(opj(bag_folders_path_meta_path, "*")), key=natural_keys)
ctr = 0
for bfp in bag_folders_paths_list:
if False: # ctr > 3:
print "Temp, returning"
return BF_dic
num_bag_files = len(gg(opj(bag_folders_path_meta_path.replace("meta", "rgb_1to4"), "*")))
run_name = fname(bfp)
for i in range(int(num_bag_files / 10)):
BF_dic_keys_weights.append(run_name)
left_image_bound_to_data_name = get_preprocess_dir_name_info(bfp)
# cprint(opj(bfp,'Bag_Folder.pkl'),'blue','on_green')
if len(gg(opj(bfp, "Bag_Folder.pkl"))) == 1:
print ("")
cprint(opj(run_name, "Bag_Folder.pkl") + " exists, loading it.", "yellow", "on_red")
BF = load_obj(opj(bfp, "Bag_Folder.pkl"))
else:
cprint("ERROR!!! " + opj(run_name, "Bag_Folder.pkl") + " does not exist!", "yellow", "on_red")
continue
assert False
if run_name in BF_dic:
cprint("ERROR!!! " + run_name + " already in BF_dic!", "yellow", "on_red")
assert False
BF_dic[run_name] = BF
ctr += 1
return BF_dic, BF_dic_keys_weights
def load_Bag_Folders(
data_path='/home/karlzipser/Desktop/bair_car_data',
ignore=['caffe', 'play', 'follow', 'furtive', 'local', 'racing']):
BagFolder_dic = {}
BagFolders_weighted = []
bag_folder_path_list_meta = sorted(gg(opj(data_path, 'meta', '*')),
key=natural_keys)
bag_folder_path_list_rgb = sorted(gg(opj(data_path, 'rgb_1to4', '*')),
key=natural_keys)
metas = []
rgbs = []
for b in bag_folder_path_list_meta:
metas.append(fname(b))
for b in bag_folder_path_list_rgb:
rgbs.append(fname(b))
for run_name in metas[:30]:
print run_name
if str_contains_one(run_name, ignore):
continue
num_bag_files = len(
gg(opj(data_path, 'rgb_1to4', run_name, '*.bag.pkl')))
left_image_bound_to_data_name = preprocess_Bag_Folders.get_preprocess_dir_name_info(
run_name)
if len(gg(opj(data_path, 'meta', run_name, 'Bag_Folder.pkl'))) == 1:
print('')
cprint(
opj(run_name, 'Bag_Folder.pkl') + ' exists, loading it.',
'yellow', 'on_red')
BagFolder = load_obj(
opj(data_path, 'meta', run_name, 'Bag_Folder.pkl'))
for i in range(int(num_bag_files / 10)):
BagFolders_weighted.append(run_name)
else:
cprint(
'Warning, ' + opj(run_name, 'Bag_Folder.pkl') +
' does not exist!', 'yellow', 'on_blue')
continue
if run_name in BagFolder_dic:
cprint('Error, ' + run_name + ' already in BagFolder!', 'yellow',
'on_blue')
assert (False)
BagFolder_dic[run_name] = BagFolder
return BagFolder_dic, BagFolders_weighted
def load_Bag_Folders(bag_folders_path_meta_path, bag_folders_rgb_1to4_path):
BF_dic = {}
BF_dic_keys_weights = []
bag_folders_paths_list = sorted(gg(opj(bag_folders_path_meta_path, '*')),
key=natural_keys)
ctr = 0
for bfp in bag_folders_paths_list:
if False: #ctr > 3:
print "Temp, returning"
return BF_dic
num_bag_files = len(
gg(opj(bag_folders_path_meta_path.replace('meta', 'rgb_1to4'),
'*')))
run_name = fname(bfp)
for i in range(int(num_bag_files / 10)):
BF_dic_keys_weights.append(run_name)
left_image_bound_to_data_name = get_preprocess_dir_name_info(bfp)
#cprint(opj(bfp,'Bag_Folder.pkl'),'blue','on_green')
if len(gg(opj(bfp, 'Bag_Folder.pkl'))) == 1:
print('')
cprint(
opj(run_name, 'Bag_Folder.pkl') + ' exists, loading it.',
'yellow', 'on_red')
BF = load_obj(opj(bfp, 'Bag_Folder.pkl'))
else:
cprint(
'ERROR!!! ' + opj(run_name, 'Bag_Folder.pkl') +
' does not exist!', 'yellow', 'on_red')
continue
assert (False)
if run_name in BF_dic:
cprint('ERROR!!! ' + run_name + ' already in BF_dic!', 'yellow',
'on_red')
assert (False)
BF_dic[run_name] = BF
ctr += 1
return BF_dic, BF_dic_keys_weights
def load_Bag_Folders(data_path='/home/karlzipser/Desktop/bair_car_data',ignore=['caffe','play','follow','furtive','local','racing']):
BagFolder_dic = {}
BagFolders_weighted = []
bag_folder_path_list_meta = sorted(gg(opj(data_path,'meta','*')),key=natural_keys)
bag_folder_path_list_rgb = sorted(gg(opj(data_path,'rgb_1to4','*')),key=natural_keys)
metas = []
rgbs = []
for b in bag_folder_path_list_meta:
metas.append(fname(b))
for b in bag_folder_path_list_rgb:
rgbs.append(fname(b))
for run_name in metas:
if str_contains_one(run_name,ignore):
continue
num_bag_files = len(gg(opj(data_path,'rgb_1to4',run_name,'*')))
left_image_bound_to_data_name = preprocess_Bag_Folders.get_preprocess_dir_name_info(run_name)
if len(gg(opj(data_path,'meta',run_name,'Bag_Folder.pkl'))) == 1:
print('')
cprint(opj(run_name,'Bag_Folder.pkl')+' exists, loading it.','yellow','on_red')
BagFolder = load_obj(opj(data_path,'meta',run_name,'Bag_Folder.pkl'))
for i in range(int(num_bag_files/10)):
BagFolders_weighted.append(run_name)
else:
cprint('Warning, '+opj(run_name,'Bag_Folder.pkl')+' does not exist!','yellow','on_blue')
continue
if run_name in BagFolder_dic:
cprint('Error, '+run_name+' already in BagFolder!','yellow','on_blue')
assert(False)
BagFolder_dic[run_name] = BagFolder
return BagFolder_dic,BagFolders_weighted
