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

add calb in dictioary definition

Browse Source
This commit is contained in:
lorenzo 2019-05-13 14:13:51 +02:00
parent 9eb4b3a5f5
commit f72be64b8d
2 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/features/preprocess_ki.py
View File

@ -66,11 +66,11 @@ class PreprocessKitti:
self.set_val = set_gt.intersection(set_val)
assert self.set_train and self.set_val, "No validation or training annotations"
self.dic_jo = {'train': dict(X=[], Y=[], names=[], kps=[],
self.dic_jo = {'train': dict(X=[], Y=[], names=[], kps=[], K=[],
clst=defaultdict(lambda: defaultdict(list))),
'val': dict(X=[], Y=[], names=[], kps=[],
'val': dict(X=[], Y=[], names=[], kps=[], K=[],
clst=defaultdict(lambda: defaultdict(list))),
'test': dict(X=[], Y=[], names=[], kps=[],
'test': dict(X=[], Y=[], names=[], kps=[], K=[],
clst=defaultdict(lambda: defaultdict(list)))}
self.dic_names = defaultdict(lambda: defaultdict(list))

2
src/predict/predict_2d_3d.py
View File

@ -1,6 +1,4 @@
"""Predict poses for given images."""
import argparse
import glob
import json
import os