활연개랑

[Tensorflow] TypeError: Input 'bytes' of 'DecodeRaw' Op has type int64 that does not match expected type of string. 본문

Python

[Tensorflow] TypeError: Input 'bytes' of 'DecodeRaw' Op has type int64 that does not match expected type of string.

승해tmdhey 2022. 4. 8. 10:52
반응형

tfrecord를 만들 때 발생하는 에러

TypeError: Input 'bytes' of 'DecodeRaw' Op has type int64 that does not match expected type of string.

 

parsing하는 과정에서 label parsing을 다음과 같이 했을 때 위 오류가 발생했다.

label = tf.io.decode_raw(features['train/label'], tf.int64)
label = tf.cast(label, tf.int64)

 

하지만, label에서는 decode_raw를 사용할 필요 없이

label = features['train/label']

이렇게만 작성해주면 된다.