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))
