• TOP
• 機械学習

[Android][TensorFlow Lite]Metadata Writer APIに定義が無い入力の定義方法

2021-09-18T09:48:00+09:00 2021-09-18T01:10:50Z 機械学習

TensorFlow Lite Metadata Writer APIには、以下の入出力用の定義(クラス)があります。

• Image classifiers
• Object detectors
• Image segmenters
• Natural language classifiers
• Audio classifiers

しかし、普通のArray(配列、ベクトル)の定義方法や例が載っておらず困りました。
結論だけ書くと以下のよう入力があることを定義すれば良いだけでした。

# 変形パラメータの配列をセット
input_param_meta = _metadata_fb.TensorMetadataT()
input_param_meta.name = "modifucation parameter"
input_param_meta.description = (
    "['Black_Hair', 'Blond_Hair', 'Brown_Hair', 'Male', 'Young']"
    "If your image is \"Young, Man, Black Hair\" = [1, 0, 0, 1, 1]"
    )
input_param_stats = _metadata_fb.StatsT()
input_param_stats.max = [1.0]
input_param_stats.min = [0.0]
input_param_meta.stats = input_param_stats