Skip to content
Snippets Groups Projects
Commit 42957cc6 authored by Stephane Chavin's avatar Stephane Chavin
Browse files

Update file get_train_val.py

parent fb5fd6eb
No related branches found
No related tags found
No related merge requests found
Show whitespace changes
Inline Side-by-side
Showing
with 1 addition and 1 deletion
get_train_val.py
+ 1
− 1
View file @ 42957cc6
...@@ -72,7 +72,7 @@ def prepare_data(arguments): ...@@ -72,7 +72,7 @@ def prepare_data(arguments):
""" """
annotations = [] annotations = []
background = [] 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='░▒▓█'): ascii='░▒▓█'):
file_annotation = pd.read_csv(f, sep=' ', names=['species', 'x', 'y', 'w', 'h']) file_annotation = pd.read_csv(f, sep=' ', names=['species', 'x', 'y', 'w', 'h'])
if len(file_annotation) == 0: if len(file_annotation) == 0:
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment