簡體   English   中英

在張量流中使用預訓練的重發模型進行對象檢測時出錯

[英]Error using the pre-trained resent model for object detection in tensorflow

我從預訓練的resnet50模型中創建了一個保存的模型,以便使用tensorflow服務模塊進行服務。 但是,當我嘗試運行模型服務器時,出現此錯誤:

2018-03-23 13:36:37.130839:E external / org_tensorflow / tensorflow / core / common_runtime / executor.cc:651]執行程序無法創建內核。 無效的參數:NodeDef在Op output:T中未提及attr“膨脹”; attr = T:type,allowed = [DT_HALF,DT_FLOAT]; attr = strides:list(int); attr = use_cudnn_on_gpu:bool,默認= true; attr = padding:string,allowed = [“ SAME”,“ VALID”]; attr = data_format:string,default =“ NHWC”,allowed = [“” NHWC“,” NCHW“]>; NodeDef:FirstStageFeatureExtractor / resnet_v1_50 / resnet_v1_50 / conv1 / Conv2D = Conv2D [T = DT_FLOAT,data_format =“ NHWC”,膨脹= [1,1,1,1],padding =“ VALID”,步幅= [1、2、2、2 ,1],use_cudnn_on_gpu = true,_device =“ / job:localhost / replica:0 / task:0 / device:CPU:0”](FirstStageFeatureExtractor / resnet_v1_50 / resnet_v1_50 / Pad,FirstStageFeatureExtractor / resnet_v1_50 / conv1 / weights / read) 。 (檢查您的GraphDef解釋二進制文件是否與生成GraphDef的二進制文件最新。) [[節點:FirstStageFeatureExtractor / resnet_v1_50 / resnet_v1_50 / conv1 / Conv2D = Conv2D [T = DT_FLOAT,data_format =“ NHWC”,膨脹= [1、1、1、1],填充=“ VALID”,步幅= [1、2 ,2,1],use_cudnn_on_gpu = true,_device =“ / job:localhost / replica:0 / task:0 / device:CPU:0”](FirstStageFeatureExtractor / resnet_v1_50 / resnet_v1_50 / Pad,FirstStageFeatureExtractor / resnet_v1_50 / conv1 / weights /讀)]]

好吧,它告訴您哪里出了問題,盡管在所有詳細信息下可能很難看到:

“檢查您的GraphDef解釋二進制文件是否與生成GraphDef的二進制文件最新”

您似乎正在使用TensorFlow的最新版本,該版本已向Conv2D添加了“ dilations”參數。

tensorflow_model_server中的TensorFlow版本比該版本早,因此不知道如何實例化該運算符。

可以嘗試在足夠老的TensorFlow版本上重建模型,也可以嘗試使用更新的TensorFlow版本創建tensorflow_model_server。

在當前的TensorFlow服務池中,這可以輕松實現

bazel build --action_env TF_REVISION="{git hash}" //tensorflow_serving/model_servers:tensorflow_model_server

有關詳細信息,請參見https://github.com/tensorflow/serving/commit/f9e602c753ef82ff96b28429dd07e900f10eb007

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM