From 9eb4b3a5f579ae412b91b038d51a99843035dfd1 Mon Sep 17 00:00:00 2001 From: lorenzo Date: Mon, 13 May 2019 13:59:31 +0200 Subject: [PATCH] add calibration in names.json --- src/features/preprocess_ki.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/src/features/preprocess_ki.py b/src/features/preprocess_ki.py index c4531fb..72b9765 100644 --- a/src/features/preprocess_ki.py +++ b/src/features/preprocess_ki.py @@ -94,6 +94,10 @@ class PreprocessKitti: self.cnt_fnf += 1 continue + # Extract keypoints + path_txt = os.path.join(self.dir_kk, basename + '.txt') + kk, tt = self.get_calibration(path_txt) + # Iterate over each line of the gt file and save box location and distances with open(path_gt, "r") as f_gt: for line_gt in f_gt: @@ -105,12 +109,9 @@ class PreprocessKitti: dds.append(dd) self.dic_names[basename + '.png']['boxes'].append(box) self.dic_names[basename + '.png']['dds'].append(dd) + self.dic_names[basename + '.png']['K'].append(kk) self.cnt_gt += 1 - # Extract keypoints - path_txt = os.path.join(self.dir_kk, basename + '.txt') - kk, tt = self.get_calibration(path_txt) - # Find the annotations if exists try: with open(os.path.join(self.dir_ann, basename + '.png.pifpaf.json'), 'r') as f: @@ -130,7 +131,7 @@ class PreprocessKitti: self.dic_jo[phase]['kps'].append(uv_kps[ii]) self.dic_jo[phase]['X'].append(inputs[ii]) self.dic_jo[phase]['Y'].append([dds[idx_max]]) # Trick to make it (nn,1) - + self.dic_jo[phase]['K'].append(kk) self.dic_jo[phase]['names'].append(name) # One image name for each annotation self.append_cluster(self.dic_jo, phase, inputs[ii], dds[idx_max], uv_kps[ii]) self.dic_cnt[phase] += 1