if __name__ == '__main__': """ Create arm model Create arm prototypes Create dataset Create eye model Create eye prototypes Combine prototypes (for the eye network) Train the network """ origin = 0 #make sure the eye and the arm have the same origin! arm = Arm(visualize = False, origin = origin) print 'Create arm proto' proto_arm = arm.create_prototypes(shape=(10, 10)) eyes = Eyes(visualize = False, origin = origin) # eyes.create_dataset(n_datapoints=100000, train_file='train_data_eyes.p', val_file='validation_data_eyes.p', test_file= 'test_data_eyes.p') print 'create eye proto' proto_eye = eyes.create_prototypes(shape = (10, 10)) print 'combine prototypes' proto_arm = combine_prototypes(proto_arm, proto_eye) print 'To the network!' train_network_scenario1(proto_eye, proto_arm, origin = origin, train_data = 'train_data_eyes.p', val_data = 'validation_data_eyes.p' ) """ origin = 0 arm = Arm(visualize= False, origin = origin) #eye.create_dataset(n_datapoints=1000000) proto = arm.create_prototypes((10,10))