Я обучил модель и набор данных изображений. Сейчас я пытаюсь запустить компонент 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
Я потратил много времени на эту проблему. Пожалуйста, помогите!
Спасибо!