diff --git a/get_time_detection.py b/get_time_detection.py index 44cd3e0596d2e2824d8cb903a52db0cfa42281eb..16f876fcd20bdbab1c7bbcd38d80b02e5870a031 100644 --- a/get_time_detection.py +++ b/get_time_detection.py @@ -23,7 +23,7 @@ out_file = str('YOLO_detection'+str('_'+str(today.day)+'_'+str(today.month))) outdir = args.directory -df = pd.concat({f:pd.read_csv(os.path.join(annots, f), sep=' ', names=['espece', 'x', 'y', 'w', 'h', 'conf']) +df = pd.concat({f:pd.read_csv(os.path.join(annots, f), sep=' ', names=['espece', 'x', 'y', 'w', 'h']) for f in tqdm(os.listdir(annots))}, names=['file']) df = df.reset_index(level=[0])