文档首页/ AI开发平台ModelArts/ 故障排除/ 推理部署/ 服务部署/ 使用AI市场物体检测YOLOv3_Darknet53算法训练后部署在线服务报错
更新时间:2023-01-10 GMT+08:00
分享

使用AI市场物体检测YOLOv3_Darknet53算法训练后部署在线服务报错

问题现象

使用AI市场物体检测YOLOv3_Darknet53算法进行训练,将数据集切分后进行部署在线服务报错,日志如下:TypeError: Cannot interpret feed_dict key as Tensor: The name 'images:0' refers to a Tensor which does not exist. The operation, 'images', does not exist in the graph。

处理方法

  • 如果切分了数据集,需要删除推理代码中“Yolov3Service”类中的如下代码:
    self.model_inputs = {'images': 'images:0'}
    self.model_outputs = {'feat_1': 'feat_1:0',
                          'feat_2': 'feat_2:0',
                          'feat_3': 'feat_3:0'}
  • 不做数据集切分操作。如果选择未切分的数据集,算法将做纯训练场景;

相关文档