Невозможно передать Evaluator с пользовательским tf.Пример - PullRequest
0 голосов
/ 03 июля 2019

Я обучил модель и набор данных изображений. Сейчас я пытаюсь запустить компонент Evaluator в TFX:

    model_analyzer = Evaluator(
        model_exports=channel.as_channel([types.TfxType(
            type_name='path/to/evalSavedModel']),
        examples=channel.as_channel([types.TfxType(examples)])
    )

где пример:

def create_examples():
    cwd = '/path/to/dataset/with/two/classes'
    keys = ['0', '1']
    values = [0, 1]
    classes = dict(zip(keys, values))
    examples = []
    for name, label in classes.items():
        class_path = os.path.join(cwd, name)
        for img_name in os.listdir(class_path):
            img_path = os.path.join(class_path, img_name)
            img = Image.open(img_path)
            img_raw = img.tobytes()
            example = tf.train.Example(features=tf.train.Features(feature={
                "img_raw": tf.train.Feature(bytes_list=tf.train.BytesList(value=[img_raw])),
                "label": tf.train.Feature(int64_list=tf.train.Int64List(value=[label]))}))
            examples.append(example)

    return examples

Однако я получаю следующую ошибку:

TypeError: [features {
  feature {
    key: "img_raw"
    value {
      bytes_list {
        value: "\234z^\233 has type list, but expected one of: bytes, unicode

Я потратил много времени на эту проблему. Пожалуйста, помогите!

Спасибо!

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...