Improved function name for SceneNet data preparation

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

View File

@ -305,7 +305,7 @@ def prepare(args: argparse.Namespace) -> None:
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)
with open(f"{args.ground_truth_path}/photo_paths.bin", "wb") as file:
pickle.dump(file_names_photos, file)

View File

@ -21,6 +21,7 @@ Functions:
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_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 Sequence
@ -361,7 +362,7 @@ def _load_images_ssd_callback(resized_shape: Sequence[int]) \
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[List[dict]]]]:
"""