2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Improved function name for SceneNet data preparation

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-13 12:05:19 +02:00
parent 58723adfe0
commit 693594e75c
2 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/twomartens/masterthesis/cli.py
View File

@ -305,7 +305,7 @@ def prepare(args: argparse.Namespace) -> None:
from twomartens.masterthesis import data from twomartens.masterthesis import data
file_names_photos, file_names_instances, instances = data.prepare_scenenet_val(args.scenenet_path, file_names_photos, file_names_instances, instances = data.prepare_scenenet_data(args.scenenet_path,
args.protobuf_path) args.protobuf_path)
with open(f"{args.ground_truth_path}/photo_paths.bin", "wb") as file: with open(f"{args.ground_truth_path}/photo_paths.bin", "wb") as file:
pickle.dump(file_names_photos, file) pickle.dump(file_names_photos, file)

3
src/twomartens/masterthesis/data.py
View File

@ -21,6 +21,7 @@ Functions:
load_coco_train(...): loads the COCO training data into a Tensorflow data set load_coco_train(...): loads the COCO training data into a Tensorflow data set
load_coco_val(...): loads the COCO validation data into a Tensorflow data set load_coco_val(...): loads the COCO validation data into a Tensorflow data set
load_scenenet_data(...): loads the SceneNet RGB-D data into a Tensorflow data set load_scenenet_data(...): loads the SceneNet RGB-D data into a Tensorflow data set
prepare_scenenet_data(...): prepares the SceneNet RGB-D data and returns it in Python format
""" """
from typing import Callable, List, Mapping, Tuple from typing import Callable, List, Mapping, Tuple
from typing import Sequence from typing import Sequence
@ -361,7 +362,7 @@ def _load_images_ssd_callback(resized_shape: Sequence[int]) \
return _load_images_ssd return _load_images_ssd
def prepare_scenenet_val(data_path: str, protobuf_path: str) -> Tuple[List[List[str]], def prepare_scenenet_data(data_path: str, protobuf_path: str) -> Tuple[List[List[str]],
List[List[str]], List[List[str]],
List[List[List[dict]]]]: List[List[List[dict]]]]:
""" """