def labSliceVideo(): fig = plt.figure(figsize=(12, 6)) fig.subplots_adjust(left=0.1, bottom=0.05, right=0.9, top=0.9, wspace=0.3, hspace=0.2) font_size = 15 fig.suptitle("Lab slice (Animation)", fontsize=font_size) plt.subplot(1, 2, 1) plt.title("OpenCV Lab2rgb", fontsize=font_size) num_slices = 101 lab_plot_cv = LabSlicePlot(LabSlice(func=Lab2rgb_cv), num_slices=num_slices) plt.subplot(1, 2, 2) plt.title("Implemented Lab2rgb", fontsize=font_size) lab_plot_py = LabSlicePlot(LabSlice(func=Lab2rgb_py), num_slices=num_slices) def animFunc(step, *args): lab_plot_cv.animationFunc(step) lab_plot_py.animationFunc(step) return figure2numpy(fig) images = [animFunc(step) for step in range(2 * num_slices)] result_file = resultFile("LabSlice", ".wmv") saveVideo(result_file, images)
def labSliceAnimation(): fig = plt.figure() fig.subplots_adjust(left=0.05, bottom=0.05, right=0.95, top=0.9, wspace=0.1, hspace=0.2) font_size = 15 fig.suptitle("Lab slice (Animation)", fontsize=font_size) plt.subplot(1, 2, 1) plt.title("OpenCV Lab2rgb", fontsize=font_size) lab_plot_cv = LabSlicePlot(LabSlice(func=Lab2rgb_cv)) plt.subplot(1, 2, 2) plt.title("Implemented Lab2rgb", fontsize=font_size) lab_plot_py = LabSlicePlot(LabSlice(func=Lab2rgb_py)) def animFunc(step, *args): plots = lab_plot_cv.animationFunc(step) plots.extend(lab_plot_py.animationFunc(step)) plt.draw() return plots ani = animation.FuncAnimation(fig, animFunc, interval=0, blit=True) showMaximize()
def __init__(self, abs_original, abs_edited, L=50, abs_animation=[]): self._slice = LabSlice(func=Lab2rgb_py) self._slice_plot = LabSlicePlot(self._slice) self._slice_plot.plot(L) self._abs_original = abs_original self._abs_edited = abs_edited self._abs_animation = abs_animation self._transfer = ABTransfer(abs_original, abs_edited) self._plot()