Commit 26af67ca authored by Gilberto Astolfi's avatar Gilberto Astolfi
Browse files

adicionando obj.data para yolo

parent ceb87c6c
......@@ -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'
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment