Character.py 872 B

123456789101112131415161718192021222324
  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. """Get the histogram of Local Binary Patterns over this entire
  10. image."""
  11. if hasattr(self, 'feature'):
  12. return
  13. pattern = LBP(self.image, neighbours=neighbours)
  14. self.feature = pattern.single_cell_features_vector()
  15. def get_feature_vector(self, cell_size=None):
  16. """Get the concatenated histograms of Local Binary Patterns. """
  17. pattern = LBP(self.image) if cell_size == None \
  18. else LBP(self.image, cell_size)
  19. return pattern.create_features_vector()