from neuralnet import NeuralNet from extractfeatures import * import cv2 if __name__ == "__main__": red = cv2.imread('/home/hoshiro/Pictures/test-img/red-light.jpg', cv2.CV_LOAD_IMAGE_COLOR) yellow = cv2.imread('/home/hoshiro/Pictures/test-img/yellow-light.jpg', cv2.CV_LOAD_IMAGE_COLOR) green = cv2.imread('/home/hoshiro/Pictures/test-img/green-light.jpg', cv2.CV_LOAD_IMAGE_COLOR) features_red = extract_hist_features(red) features_yellow = extract_hist_features(yellow) features_green = extract_hist_features(green) neural_net = NeuralNet() neural_net.build(len(features_red), len(features_red) / 2, 1) neural_net.create_data_set() neural_net.add_list_of_data([features_red], 1) neural_net.add_list_of_data([features_yellow], 2) neural_net.add_list_of_data([features_green], 3) neural_net.train() print neural_net.apply_over_data(features_yellow)