Python Image Processing Cookbook

Extended depth of field with mahotas

Perform the following steps to implement focus stacking with the mahotas library functions:

1. Create the image stack first by extracting grayscale image frames from a highway traffic video at night:

import mahotas as mhdef create_image_stack(vid_file, n = 200):  vidcap = cv2.VideoCapture(vid_file) success,image = vidcap.read() i = 0 success = True h, w = image.shape[:2] imstack = np.zeros((n, h, w)) while success and i < n:   imstack[i,...] = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)   success,image = vidcap.read()   i += 1 return imstackimage = create_image_stack('images/highway.mp4') #cloud.mp4stack,h,w = image.shape

2. Use the sobel() function from mahotas as the pixel-level measure of infocusness:

focus = np.array([mh.sobel(t, ...