• tensorflow的object-detection切割图片出现错误
?
1
2
3
4
5
6
7
8
9
10
tensorflow.python.framework.errors_impl.InvalidArgumentError: 2 root error(s) found.
  (0) Invalid argument: Key: image/object/mask.  Data types don't match. Expected type: float, Actual type: string
     [[{{node ParseSingleExample/ParseSingleExample}}]]
     [[IteratorGetNext]]
     [[BatchMultiClassNonMaxSuppression/map/while/TensorArrayReadV3_5/_7587]]
  (1) Invalid argument: Key: image/object/mask.  Data types don't match. Expected type: float, Actual type: string
     [[{{node ParseSingleExample/ParseSingleExample}}]]
     [[IteratorGetNext]]
0 successful operations.
0 derived errors ignored.
  • 错误原因

pipeline配置文件缺少参数mask_type去指定mask类型

  • 设置方法
?
1
2
3
4
5
6
7
8
train_input_reader: {
  mask_type: PNG_MASKS
}
 
 
eval_input_reader: {
  mask_type: PNG_MASKS
}