diff options
Diffstat (limited to 'canny-cv.py')
| -rw-r--r-- | canny-cv.py | 26 |
1 files changed, 13 insertions, 13 deletions
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() |