def plotDepth(image, segments, labels, gt, est, outFilename): gtImg = fillSegments(segments, gt, labels) estImg = fillSegments(segments, est, labels) gtImg = np.ma.masked_where(gtImg == 0, gtImg) estImg = np.ma.masked_where(estImg == 0, estImg) #Set range to that of the ground truth vmax = np.max(gtImg) vmin = np.min(gtImg) colormap = cm.get_cmap('jet') colormap.set_bad('black') f, ax = plt.subplots(3, 1, sharex=True) ax[0].imshow(image) ax[0].set_title("Image") ax[1].imshow(gtImg, cmap=colormap, vmax = vmax, vmin = vmin) ax[1].set_title("GT") axx=ax[2].imshow(estImg, cmap=colormap, vmax = vmax, vmin = vmin) ax[2].set_title("EST") f.subplots_adjust(right=.8) cbar_ax = f.add_axes([0.85, 0.15, 0.05, 0.7]) f.colorbar(axx, cax=cbar_ax) plt.savefig(outFilename) plt.close(f)
def plotDepth(image, segments, labels, gt, est, outFilename): gtImg = fillSegments(segments, gt, labels) estImg = fillSegments(segments, est, labels) gtImg = np.ma.masked_where(gtImg == 0, gtImg) estImg = np.ma.masked_where(estImg == 0, estImg) #Set range to that of the ground truth vmax = np.max(gtImg) vmin = np.min(gtImg) colormap = cm.get_cmap('jet') colormap.set_bad('black') f, ax = plt.subplots(3, 1, sharex=True) ax[0].imshow(image) ax[0].set_title("Image") ax[1].imshow(gtImg, cmap=colormap, vmax=vmax, vmin=vmin) ax[1].set_title("GT") axx = ax[2].imshow(estImg, cmap=colormap, vmax=vmax, vmin=vmin) ax[2].set_title("EST") f.subplots_adjust(right=.8) cbar_ax = f.add_axes([0.85, 0.15, 0.05, 0.7]) f.colorbar(axx, cax=cbar_ax) plt.savefig(outFilename) plt.close(f)
def plotImg(segments, labels, vals): img = fillSegments(segments, vals, labels) colormap = cm.get_cmap('jet') colormap.set_bad('black') img = np.ma.masked_where(img == 0, img) f = plt.imshow(img, cmap=colormap) plt.colorbar() plt.show()
def plotImg(segments, labels, vals): img = fillSegments(segments, vals, labels) colormap = cm.get_cmap('jet') colormap.set_bad('black') img = np.ma.masked_where(img == 0, img) f = plt.imshow(img, cmap=colormap) plt.colorbar() plt.show()
def plotEval(image, segments, labels, est, outFilename): estImg = fillSegments(segments, est, labels) colormap = cm.get_cmap('jet') f, ax = plt.subplots(2, 1) ax[0].imshow(image) ax[0].set_title("Image") axx=ax[1].imshow(estImg, cmap=colormap) ax[1].set_title("EST") f.subplots_adjust(right=.8) cbar_ax = f.add_axes([0.85, 0.15, 0.05, 0.7]) f.colorbar(axx, cax=cbar_ax) plt.savefig(outFilename) plt.close(f)
def plotEval(image, segments, labels, est, outFilename): estImg = fillSegments(segments, est, labels) colormap = cm.get_cmap('jet') f, ax = plt.subplots(2, 1) ax[0].imshow(image) ax[0].set_title("Image") axx = ax[1].imshow(estImg, cmap=colormap) ax[1].set_title("EST") f.subplots_adjust(right=.8) cbar_ax = f.add_axes([0.85, 0.15, 0.05, 0.7]) f.colorbar(axx, cax=cbar_ax) plt.savefig(outFilename) plt.close(f)