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()
예제 #3
0
 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()