OpenCV 4 with Python Blueprints
上QQ阅读APP看书,第一时间看更新

Mapping the GUI base class

The FilterLayout GUI will be based on a generic, plain layout class called BaseLayout, which we will be able to use in subsequent chapters as well.

The BaseLayout class is designed as an abstract base class. You can think of this class as a blueprint or recipe that will apply to all the layouts that we are yet to design, that is, a skeleton class that will serve as the backbone for all of our future GUI code.

We start the file by importing the packages that we will usethe wxPython module, which we use to create the GUI; numpy, which we use to do matrix manipulations; and OpenCV (of course):

import numpy as np
import wx
import cv2

The class is designed to be derived from the blueprint or skeleton, that is, the wx.Frame class:

class BaseLayout(wx.Frame):

Later on, when we write our own custom layout (FilterLayout), we will use the same notation to specify that the class is based on the BaseLayout blueprint (or skeleton) class, for example, in class FilterLayout(BaseLayout):. But for now, let's focus on the BaseLayout class.

An abstract class has at least one abstract method. We are going to make the method abstract by ensuring that if the method stays unimplemented, the application will not run and we throw an exception:

class BaseLayout(wx.Frame):
...
...
...
def process_frame(self, frame_rgb: np.ndarray) -> np.ndarray:
"""Process the frame of the camera (or other capture device)

:param frame_rgb: Image to process in rgb format, of shape (H, W, 3)
:return: Processed image in rgb format, of shape (H, W, 3)
"""
raise NotImplementedError()

Then, any class that is derived from it, such as FilterLayout, must specify a full implementation of that method. This will allow us to create custom layouts, as you will see in a moment.

But first, let's proceed to the GUI constructor.