edgetpu.utils.dataset_utils

Utilities to process dataset.

edgetpu.utils.dataset_utils.read_label_file(file_path)¶

Reads labels from text file and store it in a dict.

This function supports following formats:

1. Each line contains id and description separated by colon or space. Example: ‘0:cat’ or ‘0 cat’.
2. Each line contains description only. Label’s id equals to row number.

Parameters:	file_path – String, path to the label file.
Returns:	Dict of (int, string) which maps label id to description.