From e2b540c3299591e544cf9cda378b397dc4560744 Mon Sep 17 00:00:00 2001
From: Till Raab <till.raab@uni-tuebingen.de>
Date: Fri, 27 Oct 2023 12:52:37 +0200
Subject: [PATCH] suggestions for after my hollidays
---
data/generate_dataset.py | 7 +++++++
1 file changed, 7 insertions(+)
diff --git a/data/generate_dataset.py b/data/generate_dataset.py
index 2dfcd75..e72e0b6 100644
--- a/data/generate_dataset.py
+++ b/data/generate_dataset.py
@@ -172,6 +172,13 @@ def main(args):
folders = [folder for folder in folders if (folder / 'analysis' / 'rise_idx.npy').exists()]
cols = ['image', 't0', 't1', 'f0', 'f1', 'x0', 'y0', 'x1', 'y1']
bbox_df = pd.DataFrame(columns=cols)
+
+ # ToDo: Implement loading the old .csv file and upgrade it... how exactly I will determine after my vaccation
+ # bbox_df = pd.read_csv(os.path.join('dataset', 'bbox_dataset.csv'), sep=',', index_col=0)
+ # cols = list(bbox_df.keys())
+ # eval_files = []
+ # for f in pd.unique(bbox_df['image']):
+ # eval_files.append(f.split('__')[0])
else:
print('generate inference dataset ... only image output')
bbox_df = {}