diff --git a/videocr/video.py b/videocr/video.py index a3c86de..e4b7565 100644 --- a/videocr/video.py +++ b/videocr/video.py @@ -62,7 +62,7 @@ class Video: # only use bottom half of the frame by default img = img[self.height // 2:, :] # dilate and resize - img=cv2.resize(cv2.dilate(img, np.ones(2, 2), np.uint8), self.resize_dim, interpolation=cv2.INTER_AREA) + img=cv2.resize(cv2.dilate(img, np.ones((2, 2), np.uint8)), self.resize_dim, interpolation=cv2.INTER_AREA) # mask to filter out non gray-like pixels/pixels that are not bright enough hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)