deep-diver / Soccer-Ball-Detection-YOLOv2

YOLOv2 trained against custom dataset
116 stars 43 forks source link

Name error: exit is not defined while running tf.train() #10

Open HarishDechiraju opened 5 years ago

HarishDechiraju commented 5 years ago

NameError Traceback (most recent call last)

in ----> 1 tfnet.train()

~/darkflow/darkflow/net/flow.py in train(self) 38 loss_op = self.framework.loss 39 ---> 40 for i, (x_batch, datum) in enumerate(batches): 41 if not i: self.say(train_stats.format( 42 self.FLAGS.lr, self.FLAGS.batch,

~/darkflow/darkflow/net/yolo/data.py in shuffle(self) 95 def shuffle(self): 96 batch = self.FLAGS.batch ---> 97 data = self.parse() 98 size = len(data) 99

~/darkflow/darkflow/net/yolo/data.py in parse(self, exclusive) 14 if not os.path.isdir(ann): 15 msg = 'Annotation directory not found {} .' ---> 16 exit('Error: {}'.format(msg.format(ann))) 17 print('\n{} parsing {}'.format(meta['model'], ann)) 18 dumps = pascal_voc_clean_xml(ann, meta['labels'], exclusive)

NameError: name 'exit' is not

Any help?