Update file get_train_val.py

get_train_val.py

@@ -72,7 +72,7 @@ def prepare_data(arguments):
     """
     annotations = [] 
     background = []  
-    for f in tqdm(glob.glob(os.path.join(arguments.path_to_data)), desc="Processing", 
+    for f in tqdm(glob.glob(os.path.join(arguments.path_to_data, '*.txt')), desc="Processing", 
                                                             ascii='░▒▓█'):
         file_annotation = pd.read_csv(f, sep=' ', names=['species', 'x', 'y', 'w', 'h'])
         if len(file_annotation) == 0: