import os import sys from subprocess import call BASE_DIR = os.path.dirname(os.path.abspath(__file__)) sys.path.append(os.path.dirname(BASE_DIR)) from global_variables import * from utilities_common import query_yes_no if not os.path.exists(g_sun2012_data_folder): os.mkdir(g_sun2012_data_folder) # download and unzip SUN2012 data filename = os.path.join(g_sun2012_data_folder, '../SUN2012pascalformat.tar.gz') if os.path.exists(filename): if not query_yes_no( ('It seems that you have downloaded SUN2012 data to \"%s\", skip downloading?' ) % (filename)): os.remove(filename) call(['wget', '-O', filename, g_sun2012_data_url]) call([ 'tar', 'xvzf', filename, '-C', g_sun2012_data_folder, '--strip-components=1' ]) filelist = open(g_syn_bkg_filelist, 'w') for filename in sorted(os.listdir(g_syn_bkg_folder)): filelist.write(filename + '\n') filelist.close() #os.remove(filename)
return i upto += w view_distr_distance = 3 view_distr_N = 1000000 for synset in g_shapenet_synset_set: if synset not in g_view_distribution_params: print 'Please define view distribution file generation parameters for synset %s in global_variables.py' % ( synset) continue view_distr_filename = g_view_distribution_files[synset] if os.path.exists(view_distr_filename): if query_yes_no( ('It seems that you already have view distribution file \"%s\" for synset %s, skip generating?' ) % (view_distr_filename, synset)): continue else: os.remove(view_distr_filename) view_distr_azimuth_weights = g_view_distribution_params[synset][0] view_distr_elevation_weights = g_view_distribution_params[synset][1] view_distr_tilt_deviation = g_view_distribution_params[synset][2] fout = open(view_distr_filename, 'w') for _ in range(view_distr_N): azimuth_deg = weighted_choice( view_distr_azimuth_weights) * 22.5 + np.random.uniform( -11.25, 11.25) elevation_deg = weighted_choice( view_distr_elevation_weights) * 10 - 85 + np.random.uniform(-5, 5)
#!/usr/bin/python # -*- coding: utf-8 -*- import os import sys from subprocess import call BASE_DIR = os.path.dirname(os.path.abspath(__file__)) sys.path.append(os.path.dirname(os.path.dirname(BASE_DIR))) from global_variables import * from utilities_common import query_yes_no # download and unzip evaluation data url = 'https://shapenet.cs.stanford.edu/projects/JointEmbedding/data/evaluation_cross_view_image_retrieval.zip' filename = os.path.join(g_data_folder, 'evaluation_cross_view_image_retrieval.zip') if os.path.exists(filename): if query_yes_no( ('It seems that you have downloaded evaluation_cross_view_image_retrieval data to \"%s\", skip downloading?' ) % (filename)): exit() else: os.remove(filename) call(['wget', '-O', filename, url]) call(['unzip', filename, '-d', g_data_folder])
#!/usr/bin/python # -*- coding: utf-8 -*- import os import sys from subprocess import call BASE_DIR = os.path.dirname(os.path.abspath(__file__)) sys.path.append(os.path.dirname(os.path.dirname(BASE_DIR))) from global_variables import * from utilities_common import query_yes_no # download and unzip evaluation data url = 'https://shapenet.cs.stanford.edu/projects/JointEmbedding/data/evaluation_cross_view_image_retrieval.zip' filename = os.path.join(g_data_folder, 'evaluation_cross_view_image_retrieval.zip') if os.path.exists(filename): if query_yes_no(('It seems that you have downloaded evaluation_cross_view_image_retrieval data to \"%s\", skip downloading?')%(filename)): exit() else: os.remove(filename) call(['wget', '-O', filename, url]) call(['unzip', filename, '-d', g_data_folder])
#!/usr/bin/python # -*- coding: utf-8 -*- import os import sys from subprocess import call BASE_DIR = os.path.dirname(os.path.abspath(__file__)) sys.path.append(os.path.dirname(BASE_DIR)) from global_variables import * from utilities_common import query_yes_no if not os.path.exists(g_sun2012_data_folder): os.mkdir(g_sun2012_data_folder) # download and unzip SUN2012 data filename = os.path.join(g_sun2012_data_folder, '../SUN2012pascalformat.tar.gz') if os.path.exists(filename): if not query_yes_no(('It seems that you have downloaded SUN2012 data to \"%s\", skip downloading?')%(filename)): os.remove(filename) call(['wget', '-O', filename, g_sun2012_data_url]) call(['tar', 'xvzf', filename, '-C', g_sun2012_data_folder, '--strip-components=1']) filelist = open(g_syn_bkg_filelist, 'w') for filename in sorted(os.listdir(g_syn_bkg_folder)): filelist.write(filename+'\n') filelist.close() #os.remove(filename)
for i, w in enumerate(weights): if upto + w > r: return i upto += w view_distr_distance = 3 view_distr_N = 1000000 for synset in g_shapenet_synset_set: if synset not in g_view_distribution_params: print 'Please define view distribution file generation parameters for synset %s in global_variables.py'%(synset) continue view_distr_filename = g_view_distribution_files[synset] if os.path.exists(view_distr_filename): if query_yes_no(('It seems that you already have view distribution file \"%s\" for synset %s, skip generating?')%(view_distr_filename, synset)): continue else: os.remove(view_distr_filename) view_distr_azimuth_weights = g_view_distribution_params[synset][0] view_distr_elevation_weights = g_view_distribution_params[synset][1] view_distr_tilt_deviation = g_view_distribution_params[synset][2] fout = open(view_distr_filename,'w') for _ in range(view_distr_N): azimuth_deg = weighted_choice(view_distr_azimuth_weights) * 22.5 + np.random.uniform(-11.25, 11.25) elevation_deg = weighted_choice(view_distr_elevation_weights) * 10 - 85 + np.random.uniform(-5,5) tilt_deg = np.random.normal(0,view_distr_tilt_deviation) distance = view_distr_distance fout.write('%f %f %f %f\n' % (azimuth_deg, elevation_deg, tilt_deg, distance)) fout.close()