Character.py 710 B

123456789101112131415161718192021
  1. from LocalBinaryPatternizer import LocalBinaryPatternizer as LBP
  2. class Character:
  3. def __init__(self, value, corners, image, filename=None):
  4. self.value = value
  5. self.corners = corners
  6. self.image = image
  7. self.filename = filename
  8. def get_single_cell_feature_vector(self, neighbours=5):
  9. if hasattr(self, 'feature'):
  10. return
  11. pattern = LBP(self.image, neighbours=neighbours)
  12. self.feature = pattern.single_cell_features_vector()
  13. def get_feature_vector(self, cell_size=None):
  14. pattern = LBP(self.image) if cell_size == None \
  15. else LBP(self.image, cell_size)
  16. return pattern.create_features_vector()