使用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'}
- 不做数据集切分操作。如果选择未切分的数据集,算法将做纯训练场景;