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

Update file get_spectrogram.py

parent f23c146a
Branches HighBlueRec
No related tags found
No related merge requests found
...@@ -84,7 +84,7 @@ if __name__ == "__main__": ...@@ -84,7 +84,7 @@ if __name__ == "__main__":
df = pd.read_csv(args.file, low_memory=False) df = pd.read_csv(args.file, low_memory=False)
df['Path'] = df[args.columns_name] df['Path'] = df[args.columns_name]
elif args.input == 'folder': elif args.input == 'folder':
if args.file != 'None' if args.file != 'None':
df = pd.DataFrame(glob.glob(os.path.join(path_to_data, args.file), recursive=True), columns=['Path']) df = pd.DataFrame(glob.glob(os.path.join(path_to_data, args.file), recursive=True), columns=['Path'])
else : else :
df = pd.DataFrame(glob.glob(os.path.join(path_to_data, '*'), recursive=True), columns=['Path']) df = pd.DataFrame(glob.glob(os.path.join(path_to_data, '*'), recursive=True), columns=['Path'])
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment