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 colorTransferVideo(figsize=(12, 6)):
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("Color Transfer (Animation)", fontsize=font_size)
ab_original = [(-10, -20), (-20, 10), (40, 20)]
ab_edited = [(30, -50), (10, 30), (60, -20)]
abs_points = [(0, 0), (-20, 10), (0, -30), (-20, 10), (-20, 20), (40, 20), (40, -10)]
abs_animation = resample(abs_points, num=100)
transfer_plot = ABTransferPlot(ab_original, ab_edited, abs_animation=abs_animation)
def animFunc(step, *args):
transfer_plot.animationFunc(step)
return figure2numpy(fig)
images = [animFunc(step) for step in range(2*len(abs_animation))]
result_file = resultFile("ColorTransfer", ".wmv")
saveVideo(result_file, images)
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)
