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

fix names of splits

Browse Source
This commit is contained in:
lorenzo 2019-05-13 13:55:57 +02:00
parent cb344c4664
commit 3235e9d138
4 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
README.md
View File

@ -112,7 +112,8 @@ data/kitti/images`
Download nuScenes dataset (any version: Mini, Teaser or TrainVal) from [nuScenes](https://www.nuscenes.org/download), Download nuScenes dataset (any version: Mini, Teaser or TrainVal) from [nuScenes](https://www.nuscenes.org/download),
save it anywhere and soft link it in `data/nuscenes` save it anywhere and soft link it in `data/nuscenes`
### 3) Train Val Splits
Fo
### Input joints for training ### Input joints for training
MonoLoco is trained using 2D human pose joints detected by pifpaf and matched with the ground truth location provided by MonoLoco is trained using 2D human pose joints detected by pifpaf and matched with the ground truth location provided by

4
src/eval/run_kitti.py
View File

@ -23,8 +23,8 @@ class RunKitti:
self.n_samples = 100 self.n_samples = 100
list_ann = glob.glob(os.path.join(dir_ann, '*.json')) list_ann = glob.glob(os.path.join(dir_ann, '*.json'))
self.dir_kk = os.path.join('data', 'baselines', 'calib') self.dir_kk = os.path.join('data', 'kitti', 'calib')
self.dir_out = os.path.join('data', 'baselines', 'monoloco') self.dir_out = os.path.join('data', 'kitti', 'monoloco')
if not os.path.exists(self.dir_out): if not os.path.exists(self.dir_out):
os.makedirs(self.dir_out) os.makedirs(self.dir_out)
print("Created output directory for txt files") print("Created output directory for txt files")

6
src/features/preprocess_ki.py
View File

@ -17,7 +17,7 @@ class PreprocessKitti:
self.iou_thresh = iou_thresh self.iou_thresh = iou_thresh
self.dir_gt = os.path.join('data', 'kitti', 'gt') self.dir_gt = os.path.join('data', 'kitti', 'gt')
self.names_gt = os.listdir(self.dir_gt) self.names_gt = os.listdir(self.dir_gt)
self.dir_kk = os.path.join('data', 'baselines', 'calib') self.dir_kk = os.path.join('data', 'kitti', 'calib')
self.list_gt = glob.glob(self.dir_gt + '/*.txt') self.list_gt = glob.glob(self.dir_gt + '/*.txt')
assert os.path.exists(self.dir_gt), "Ground truth dir does not exist" assert os.path.exists(self.dir_gt), "Ground truth dir does not exist"
assert os.path.exists(self.dir_ann), "Annotation dir does not exist" assert os.path.exists(self.dir_ann), "Annotation dir does not exist"
@ -27,8 +27,8 @@ class PreprocessKitti:
dir_out = os.path.join('data', 'arrays') dir_out = os.path.join('data', 'arrays')
self.path_joints = os.path.join(dir_out, 'joints-kitti-' + now_time + '.json') self.path_joints = os.path.join(dir_out, 'joints-kitti-' + now_time + '.json')
self.path_names = os.path.join(dir_out, 'names-kitti-' + now_time + '.json') self.path_names = os.path.join(dir_out, 'names-kitti-' + now_time + '.json')
path_train = os.path.join('data', 'baselines', 'train.txt') path_train = os.path.join('splits', 'kitti_train.txt')
path_val = os.path.join('data', 'baselines', 'val.txt') path_val = os.path.join('splits', 'kitti_val.txt')
logging.basicConfig(level=logging.INFO) logging.basicConfig(level=logging.INFO)
self.logger = logging.getLogger(__name__) self.logger = logging.getLogger(__name__)

4
src/features/preprocess_nu.py
View File

@ -73,11 +73,11 @@ class PreprocessNuscenes:
elif dataset == 'nuscenes_teaser': elif dataset == 'nuscenes_teaser':
self.nusc = NuScenes(version='v1.0-trainval', dataroot=dir_nuscenes, verbose=True) self.nusc = NuScenes(version='v1.0-trainval', dataroot=dir_nuscenes, verbose=True)
with open("data/baselines/teaser_scenes.txt", "r") as ff: with open("splits/nuscenes_teaser_scenes.txt", "r") as ff:
teaser_scenes = ff.read().splitlines() teaser_scenes = ff.read().splitlines()
self.scenes = self.nusc.scene self.scenes = self.nusc.scene
self.scenes = [scene for scene in self.scenes if scene['token'] in teaser_scenes] self.scenes = [scene for scene in self.scenes if scene['token'] in teaser_scenes]
with open("data/baselines/split_teaser_scenes.json", "r") as ff: with open("data/splits/split_nuscenes_teaser.json", "r") as ff:
dic_split = json.load(ff) dic_split = json.load(ff)
self.split_train = [scene['name'] for scene in self.scenes if scene['token'] in dic_split['train']] self.split_train = [scene['name'] for scene in self.scenes if scene['token'] in dic_split['train']]
self.split_val = [scene['name'] for scene in self.scenes if scene['token'] in dic_split['val']] self.split_val = [scene['name'] for scene in self.scenes if scene['token'] in dic_split['val']]