plt.imshow(np.array(Image.open(image_files2[1])), vmin=0, vmax=255) plt.title(image_files2[1]) plt.show() img1_color = np.array(Image.open(image_files1[0])) img2_color = np.array(Image.open(image_files2[0])) num_bins_color = 30 hist1_rgb = histogram_module.rgb_hist(img1_color.astype('double'), num_bins_color) hist2_rgb = histogram_module.rgb_hist(img2_color.astype('double'), num_bins_color) print('Distance according to dist_l2 %f' % dist_module.dist_l2(hist1_rgb, hist2_rgb)) print('Distance according to dist_interset %f' % dist_module.dist_intersect(hist1_rgb, hist2_rgb)) print('Distance according to dist_chi2 %f' % dist_module.dist_chi2(hist1_rgb, hist2_rgb)) distance_types = ['l2', 'intersect', 'chi2'] print('distance functions:', distance_types) hist_types = ['grayvalue', 'rgb', 'rg', 'dxdy'] print('histogram types:', hist_types) num_bins_color = 30 num_bins_gray = 90 # %% for img1_file in image_files1:
image_files2 = ['./model/obj91__0.png', './model/obj94__0.png'] plt.figure() plt.subplot(1,3,1); plt.imshow(np.array(Image.open(image_files1[0])), vmin=0, vmax=255); plt.title(image_files1[0]) plt.subplot(1,3,2); plt.imshow(np.array(Image.open(image_files2[0])), vmin=0, vmax=255); plt.title(image_files2[0]) plt.subplot(1,3,3); plt.imshow(np.array(Image.open(image_files2[1])), vmin=0, vmax=255); plt.title(image_files2[1]) plt.show() img1_color = np.array(Image.open(image_files1[0])) img2_color = np.array(Image.open(image_files2[0])) num_bins_color = 30 hist1_rgb = histogram_module.rgb_hist(img1_color.astype('double'), num_bins_color) hist2_rgb = histogram_module.rgb_hist(img2_color.astype('double'), num_bins_color) print('Distance according to dist_l2 %f'% dist_module.dist_l2(hist1_rgb,hist2_rgb)) print('Distance according to dist_interset %f'% dist_module.dist_intersect(hist1_rgb,hist2_rgb)) print('Distance according to dist_chi2 %f'% dist_module.dist_chi2(hist1_rgb,hist2_rgb)) distance_types = ['l2', 'intersect', 'chi2'] print('distance functions:', distance_types) hist_types = ['grayvalue', 'rgb', 'rg', 'dxdy'] print('histogram types:', hist_types) num_bins_color = 30 num_bins_gray = 90 for img1_file in image_files1: img1_color = np.array(Image.open(img1_file)) img1_color = img1_color.astype('double')