charles/monoloco
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix dic_gt

Browse Source
This commit is contained in:
Lorenzo 2021-03-22 15:07:23 +01:00
parent 5aee21743a
commit 6d775a338b
2 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
monoloco/network/process.py
View File

@ -77,16 +77,17 @@ def factory_for_gt(im_size, focal_length=5.7, name=None, path_gt=None):
dic_gt = dic_names[name]
# Without ground-truth-file
else:
if im_size[0] / im_size[1] > 2.5: # KITTI default
elif im_size[0] / im_size[1] > 2.5: # KITTI default
kk = [[718.3351, 0., 600.3891], [0., 718.3351, 181.5122], [0., 0., 1.]] # Kitti calibration
dic_gt = None
logger.info("Using KITTI calibration matrix...")
else: # nuScenes camera parameters
kk = [
[im_size[0]*focal_length/Sx, 0., im_size[0]/2],
[0., im_size[1]*focal_length/Sy, im_size[1]/2],
[0., 0., 1.]]
logger.info("Using a standard calibration matrix...")
else: # nuScenes camera parameters
kk = [
[im_size[0]*focal_length/Sx, 0., im_size[0]/2],
[0., im_size[1]*focal_length/Sy, im_size[1]/2],
[0., 0., 1.]]
dic_gt = None
logger.info("Using a standard calibration matrix...")
return kk, dic_gt

1
monoloco/visuals/printer.py
View File

@ -59,6 +59,7 @@ class Printer:
self.kk = kk
self.output_types = args.output_types
self.z_max = args.z_max # set max distance to show instances
self.show = args.show
self.show_all = args.show_all
self.save = not args.no_save