def main():
search_id_pair= [[3501,3845],[9395,12209],[90058,91110],[126257,126638],[129707, 130171]] #Find the image id you decided to use
data_path = 'val2014'
save_path = 'test_data/merge_seam_data/simple_merge/'
for id_pair in search_id_pair:
id1, id2 = id_pair
im1 = cocotools.get_image(data_path,id1)
im2 = cocotools.get_image(data_path,id2)
combined_im = combine_horizontal(im1,im2)
file_path = os.path.join(save_path,cocotools.get_filename(id1,False) + '_' + str(id2).zfill(12) + '.jpg')
plt.imsave(file_path, combined_im)
all_imname = cocotools.get_all_images(data_path)
generate_random = 10
for i in range(generate_random):
id1 = cocotools.get_cocoid(random.choice(all_imname))
id2 = cocotools.get_cocoid(random.choice(all_imname))
im1 = cocotools.get_image(data_path, id1)
im2 = cocotools.get_image(data_path, id2)
combined_im = combine_horizontal(im1,im2)
file_path = os.path.join(save_path,cocotools.get_filename(id1,False) + '_' + str(id2).zfill(12) + '_r.jpg')
plt.imsave(file_path, combined_im)
#Chosen image reference
#FOOD 3501 , 3690, 3845
#BASEDBALL 9395, 12209
#COLOR MATCH BUT WEIRD 90058 91110
#JUST WEIRD 126257, 126638
#JUST NO 129707, 130171
return
def main():
data_path = 'val2014'
save_path = 'test_data/cross_dissolve_data'
all_imname = cocotools.get_all_images(data_path)
generate_random = 10
for i in range(generate_random):
id1 = cocotools.get_cocoid(random.choice(all_imname))
id2 = cocotools.get_cocoid(random.choice(all_imname))
im1 = cocotools.get_image_plt(data_path, id1)
im2 = cocotools.get_image_plt(data_path, id2)
if np.shape(im1) != np.shape(im2):
print "Shape does not match"
im1, im2 = cocotools.reshape_crop(im1,im2)
file_path = os.path.join(save_path,cocotools.get_filename(id1,False) + '_' + str(id2).zfill(12))
generate_cross_dissolve(im1,im2, file_path)
return
