tensorflow2 中 tf.data.Dataset.map中利用tf.py_function() 问题
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
原创
2020-09-17 18:43:33 ·
1044 阅读 ·
0 评论