Как экспортировать пользовательскую модель из TensorflowJS в Tensorflow Python

0

Вопрос

Я пытаюсь экспортировать пользовательскую модель, созданную с помощью TensorflowJS, в Tensorflow Python. Главная проблема заключается в том, что TF.JS сохраняет два артефакта: "модель-js.json" и "модель-js.weights.bin"

Я попытался импортировать JSON:

    with open('model-js/model-js.json') as json_file:
        json_config = str(json.loads(json_file.read())["modelTopology"])
    
    json_config=json_config.replace("'","\"")

    new_model = tf.keras.models.model_from_json(json_config)

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

~\anaconda3\lib\json\decoder.py in raw_decode(self, s, idx)
    353             obj, end = self.scan_once(s, idx)
    354         except StopIteration as err:
--> 355             raise JSONDecodeError("Expecting value", s, err.value) from None
    356         return obj, end

Итак, строка 103 в файле Json-это "N" в "Нуле" :

    "config": {
        "batch_input_shape": [
            null,
            128,
            128,
            1
         ],
         "dtype": "float32",
         "sparse": false,
         "name": "conv2d_input"
    }

Это позже будет изменено "Null" на "None" при импорте JSON в python.

Почему артефакты модели TensorflowJS не вписываются в Python Tensorflow?

Другое дело, что я не знаю, как импортировать веса в файл .bin. Есть какие - то другие способы сделать это? Или существуют другие типы файлов, которые можно экспортировать из TF.JS?

Заранее спасибо

javascript keras python tensorflow
2021-11-23 10:51:24
1

Лучший ответ

0

Форматы совершенно разные, это никогда так не сработает

И нет официального способа преобразовать графовую модель TFJS в сохраненную модель TF, так как преобразование происходит в одну сторону в противоположном направлении

Возможно, вам повезет с использованием стороннего инструмента преобразования, ознакомьтесь https://github.com/patlevin/tfjs-to-tf

2021-12-13 17:40:32

На других языках

Эта страница на других языках

Italiano
..................................................................................................................
Polski
..................................................................................................................
Română
..................................................................................................................
한국어
..................................................................................................................
हिन्दी
..................................................................................................................
Français
..................................................................................................................
Türk
..................................................................................................................
Česk
..................................................................................................................
Português
..................................................................................................................
ไทย
..................................................................................................................
中文
..................................................................................................................
Español
..................................................................................................................
Slovenský
..................................................................................................................