VGG框架搭建时遇到的坑1
报错内容如下图所示:
找了很多原因,修改了几次框架,之后发现是cast没有办法将str类型转换为int,所以在输入的时候直接修改了label的类型,之后运行正常了。
研究了一下,是否先转tensor再使用cast会可行,but失败,
使用to_int32也失败,所以还是输入的时候处理好吧!
label = ['0', '1', '0', '0', '1']
1、label = tf.cast(label,tf.int32)
2、label = tf.to_int32(label)