From 0ad66a09aaf55023d8eaff70094b0798b01adb19 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Sat, 12 May 2018 18:02:32 +0200 Subject: mogrify script, flow scripts, dataset options --- canny-cv.py | 26 +++++++++++++------------- 1 file changed, 13 insertions(+), 13 deletions(-) (limited to 'canny-cv.py') diff --git a/canny-cv.py b/canny-cv.py index f8b5dc3..c9ece6b 100644 --- a/canny-cv.py +++ b/canny-cv.py @@ -69,6 +69,10 @@ if __name__ == '__main__': edges_path = render_dir + "frame_{:05d}.png".format(i+1) render_path = render_dir + "ren_{:05d}.png".format(i+1) + image_pil = Image.fromarray(im, mode='RGB') + image_pil.save(tmp_path) + os.rename(tmp_path, render_path) + if dataset.name() == 'RecursiveDatasetDataLoader': # print(visuals.keys()) # s = 256 @@ -90,19 +94,15 @@ if __name__ == '__main__': tmp_im = im.copy().astype('uint8') #print(im.shape, im.dtype) - image_pil = Image.fromarray(tmp_im, mode='RGB') - image_pil.save(tmp_path) - os.rename(tmp_path, render_path) - - image_pil = Image.fromarray(im, mode='RGB') - image_pil = crop_image(image_pil, (0.50, 0.50), 0.5) - im = np.asarray(image_pil).astype('uint8') - #print(im.shape, im.dtype) - opencv_image = im[:, :, ::-1].copy() - opencv_image = cv2.GaussianBlur(opencv_image, (blur,blur), sigma) - opencv_image = cv2.Canny(opencv_image, canny_lo, canny_hi) - cv2.imwrite(tmp_path, opencv_image) - os.rename(tmp_path, edges_path) + image_pil = Image.fromarray(im, mode='RGB') + image_pil = crop_image(image_pil, (0.50, 0.50), 0.5) + im = np.asarray(image_pil).astype('uint8') + #print(im.shape, im.dtype) + opencv_image = im[:, :, ::-1].copy() + opencv_image = cv2.GaussianBlur(opencv_image, (blur,blur), sigma) + opencv_image = cv2.Canny(opencv_image, canny_lo, canny_hi) + cv2.imwrite(tmp_path, opencv_image) + os.rename(tmp_path, edges_path) webpage.save() -- cgit v1.2.3-70-g09d2