adicionando obj.data para yolo

yolo_utils.py

@@ -90,7 +90,7 @@ class YoloUtils:
 					path_yolo_txt = fold_src + '/' + set_+'/' + fold +'_'+set_+'.txt'
 					self.createFileSetYolo(path_set, path_yolo_txt)
 
-					if set_ != 'train': 
+					if set_ == 'train': 
 						path_train = path_yolo_txt
 						num_classes = len(dictClasses)
 					else: 
@@ -130,6 +130,9 @@ class YoloUtils:
 		for fold in os.listdir(repositorio):
 			fold_src = repositorio +'/'+fold
 			for set_ in os.listdir(fold_src):
+				path = fold_src + '/' + set_
+				if not os.path.isdir(path):
+					continue
 				if set_ != 'test':
 					input_names = fold_src + '/' + set_+'/' + fold +'_'+set_+'.names'
 					input_csv = path_tfrecords +'/'+fold + '/' + set_+'/' + fold +'_'+set_+'.csv'