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()
예제 #4
0
class ABTransferPlot:
    ## Constructor
    #  @param abs_original     original ab coordinates.
    #  @param abs_edited       edited ab coordinates.
    #  @param L                target L coordinate.
    #  @param abs_animation    list of ab coordinates for plot animation.
    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()

    ## Animation function for matplot.
    def animationFunc(self, step, *args):
        ab_id = step % len(self._abs_animation)

        ab_original = self._abs_animation[ab_id]
        xy_original, xy_edited = self._blendResult(ab_original)
        self._setArrow(self._blend_plot, xy_original, xy_edited)

        return self._blend_plot

    def _plot(self):
        xys_original = [self._slice.ab2xy(ab_original) for ab_original in self._abs_original]
        xys_edited = [self._slice.ab2xy(ab_edited) for ab_edited in self._abs_edited]
        for xy_original, xy_edited in zip(xys_original, xys_edited):
            self._arrow(xy_original, xy_edited)

        xy_original, xy_edited = self._blendResult(self._abs_animation[0])
        self._blend_plot = self._arrow(xy_original, xy_edited, color=[0.7, 0.5, 0.4])

    def _arrow(self, ps, pe, color=[1, 1, 1]):
        xs = [ps[0], pe[0]]
        ys = [ps[1], pe[1]]
        return [
            plt.plot(xs, ys, "-", color=color, linewidth=2, alpha=0.8)[0],
            plt.plot(ps[0], ps[1], "o", color=color, linewidth=2, alpha=0.8)[0],
        ]

    def _setArrow(self, arrow_plot, ps, pe):
        xs = [ps[0], pe[0]]
        ys = [ps[1], pe[1]]

        arrow_plot[0].set_data(xs, ys)
        arrow_plot[1].set_data(ps[0], ps[1])

    def _blendResult(self, ab_original):
        ab_edited = self._transfer.transfer(ab_original)
        xy_original = self._slice.ab2xy(ab_original)
        xy_edited = self._slice.ab2xy(ab_edited)
        return xy_original, xy_edited
예제 #5
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()
예제 #6
0
class ABTransferPlot:
    ## Constructor
    #  @param abs_original     original ab coordinates.
    #  @param abs_edited       edited ab coordinates.
    #  @param L                target L coordinate.
    #  @param abs_animation    list of ab coordinates for plot animation.
    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()

    ## Animation function for matplot.
    def animationFunc(self, step, *args):
        ab_id = step % len(self._abs_animation)

        ab_original = self._abs_animation[ab_id]
        xy_original, xy_edited = self._blendResult(ab_original)
        self._setArrow(self._blend_plot, xy_original, xy_edited)

        return self._blend_plot

    def _plot(self):
        xys_original = [
            self._slice.ab2xy(ab_original)
            for ab_original in self._abs_original
        ]
        xys_edited = [
            self._slice.ab2xy(ab_edited) for ab_edited in self._abs_edited
        ]
        for xy_original, xy_edited in zip(xys_original, xys_edited):
            self._arrow(xy_original, xy_edited)

        xy_original, xy_edited = self._blendResult(self._abs_animation[0])
        self._blend_plot = self._arrow(xy_original,
                                       xy_edited,
                                       color=[0.7, 0.5, 0.4])

    def _arrow(self, ps, pe, color=[1, 1, 1]):
        xs = [ps[0], pe[0]]
        ys = [ps[1], pe[1]]
        return [
            plt.plot(xs, ys, '-', color=color, linewidth=2, alpha=0.8)[0],
            plt.plot(ps[0], ps[1], 'o', color=color, linewidth=2, alpha=0.8)[0]
        ]

    def _setArrow(self, arrow_plot, ps, pe):
        xs = [ps[0], pe[0]]
        ys = [ps[1], pe[1]]

        arrow_plot[0].set_data(xs, ys)
        arrow_plot[1].set_data(ps[0], ps[1])

    def _blendResult(self, ab_original):
        ab_edited = self._transfer.transfer(ab_original)
        xy_original = self._slice.ab2xy(ab_original)
        xy_edited = self._slice.ab2xy(ab_edited)
        return xy_original, xy_edited