bincam = BinaryCamera('./meta.txt') bincam.open() options = AMTOptions() #t = TurnTableControl() # the com number may need to be changed. Default of com7 is used #izzy = PyControl(115200, .04, [0,0,0,0,0],[0,0,0]) # same with this c = None #c = XboxController([options.scales[0],155,options.scales[1],155,options.scales[2],options.scales[3]]) izzy = DexRobotZeke() izzy._zeke.steady(False) t = DexRobotTurntable() #options.tf_net = net5.NetFive() #options.tf_net_path = '/home/annal/Izzy/vision_amt/Net/tensor/net5/net5_02-15-2016_11h58m56s.ckpt' options.tf_net = net6.NetSix() #options.tf_net_path = '/media/1tb/Izzy/nets/net6_02-26-2016_17h58m15s.ckpt' #options.tf_net_path = '/media/1tb/Izzy/nets/net6_02-27-2016_15h30m01s.ckpt' #options.tf_net_path = '/media/1tb/Izzy/nets/net6_03-12-2016_15h03m44s.ckpt' options.tf_net_path = '/media/1tb/Izzy/nets/net6_03-27-2016_12h04m13s.ckpt' amt = AMT(bincam, izzy, t, c, options=options) while True: print "Waiting for keypress ('q' -> quit, 'r' -> rollout, 'u' -> update weights, 't' -> test, 'd' -> demonstrate, 'c' -> compile train/test sets): " char = getch() if char == 'q': print "Quitting..." break elif char == 'r':
from Net.tensor import net3 from Net.tensor import net4 from Net.tensor import net6, net8 from Net.tensor import inputdata from options import AMTOptions data = inputdata.AMTData(AMTOptions.train_file, AMTOptions.test_file, channels=3) net = net6.NetSix() # path = '/media/1tb/Izzy/nets/net6_03-27-2016_11h16m08s.ckpt' net.optimize(4000, data, batch_size=200)
def __init__(self, net_path): self.tf_net = net6.NetSix() self.tf_net_path = net_path self.sess = self.tf_net.load(var_path=self.tf_net_path) self.vals = []