как я могу изменить все соответствующие экземпляры uint8 на int8 в моей модели тензорного потока

Question

Я конвертирую замороженную модель pb в onnx, но onnx не поддерживает uint8. Как я go о замене uint8 на int8 в файле модели? Я также могу переобучиться, если придется, но я не уверен, какой файл изменить и как. Я бы предположил, что мне нужно изменить файл, который содержит нейронную архитектуру net: https://github.com/tensorflow/models/blob/master/research/slim/nets/mobilenet/mobilenet_v2.py

, но это не так ...

Answer 1

В зоопарке модели ONNX уже есть Mobil eNet v2 Модель ONNX , которая уже была преобразована (хотя исходная модель была первоначально из m xnet). Это может быть полезно.

Операторы ONNX поддерживают типы данных uint8, как описано в документации ONNX IR .