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()
