当前位置: 代码迷 >> 综合 >> tensorflow2 中 tf.data.Dataset.map中利用tf.py_function() 问题
  详细解决方案

tensorflow2 中 tf.data.Dataset.map中利用tf.py_function() 问题

热度:42   发布时间:2024-01-05 06:25:01.0

       TF中经常会遇到数据预处理时tf自带API无法满足要求只能利用numpy或者CV2去处理数据,这个时候tf.data.Dataset.map的function参数就要求用tf.py_function封装。

       但是tf.py_function需要提供inp和Tout参数,测试发现如果输入是tf.string类型map一直报错,这个时候只需要加上lambda就搞定了就不需要你强制告诉tf.py_function数据输入类型是什么了。

 

tf.data.Dataset.from_tensor_slices(all_image_path).map(lambda x:tf.py_function(load_and_preprocess_image, inp=[x],Tout=tf.float32))

 

  相关解决方案