コード例 #1
0
ファイル: formula.py プロジェクト: Airflame/mathreader
 def load(self, filename) -> None:
     """
     Loads an image containing a mathematical expression and extracts individual symbols from it
     @param filename: Path to a file
     """
     self.image = cv2.imread(filename)
     self.segments = Processing.extract_segments(self.image,
                                                 draw_rectangles=True)
コード例 #2
0
 def load(self) -> None:
     """
     Loads fonts from data/ folder and creates data and labels used for training the network
     """
     for font in range(self.fonts):
         image = cv2.imread('data/font' + str(font) + '.png')
         self.data.extend(Processing.extract_segments(image))
     self.labels = [[0] * len(Constants.symbols)
                    for _ in range(len(Constants.symbols))]
     for i in range(len(self.labels)):
         self.labels[i][i] = 1
     self.labels *= self.fonts