Обновление обученных моделей обнаружения объектов в соответствии с обновлениями TF - PullRequest
0 голосов
/ 26 марта 2019

Я перехожу на новую версию TF по соображениям стабильности (я использовал ночную сборку докера на Ubuntu 18.04 до перехода основной линии на CUDA 10). Когда я пытаюсь запустить свои модели в новой версии, я получаю следующую ошибку, которая, как я предполагаю, означает несовместимость с моделями, которые обучались на более старой версии.
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/framework/importer.py", line 426, in import_graph_def graph._c_graph, serialized, options) # pylint: disable=protected-access tensorflow.python.framework.errors_impl.InvalidArgumentError: NodeDef mentions attr 'explicit_paddings' not in Op<name=Conv2D; signature=input:T, filter:T -> output:T; attr=T:type,allowed=[DT_HALF, DT_BFLOAT16, DT_FLOAT, DT_DOUBLE]; attr=strides:list(int); attr=use_cudnn_on_gpu:bool,default=true; attr=padding:string,allowed=["SAME", "VALID"]; attr=data_format:string,default="NHWC",allowed=["NHWC", "NCHW"]; attr=dilations:list(int),default=[1, 1, 1, 1]>; NodeDef: {{node FirstStageFeatureExtractor/resnet_v1_101/resnet_v1_101/conv1/Conv2D}}. (Check whether your GraphDef-interpreting binary is up to date with your GraphDef-generating binary.).

Что мне нужно сделать, чтобы обновить ранее обученные модели для работы с новой версией TF, или мне нужно продолжать работать с этой версией до моей следующей тренировки?

1 Ответ

0 голосов
/ 27 марта 2019

После некоторого просмотра, график должен быть обновлен. Поскольку у меня еще не было контрольных точек обучения, я успешно обновил график, экспортировав из ранее замороженного графика в качестве контрольной точки. python3 export_inference_graph.py --input_type image_tensor --pipeline_config_path FROZENGRAPHDIRECTORY/pipeline.config --trained_checkpoint_prefix FROZENGRAPHDIRECTORY/model.ckpt --output_directory FROZENGRAPHDIRECTORY_tfNEWTFVERSION

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