Listelerdeki TFRecord özelliklerini nasıl yazıyorsunuz?

Question

Eğitim örneği başına N sınıflandırma etiketlerini alan bir CNN modelim var ve veri kümemden int64s listesi olan bir etiket özelliğine sahip TFRecords oluşturmaya çalışıyorum.

Yığın oluşturma tarafında, aşağıdaki gibi bir şey kullanıyorum. Ben kodunda açıkça etiket verilerini koyduk ama belli ki her bir numune için farklı olacaktır: okuma tarafında

example = tf.train.Example(features=tf.train.Features(feature={ 
    ... # other stuff 
    'label': tf.train.Feature(
        int64_list=tf.train.Int64List(value=[1, 2, 3, 4])})) 
writer.write(example.SerializeToString()) 

, ben aşağıdaki gibi bir şey yapıyorum. Ben etiket sabit sayıda varsayalım (4): Bu Tensorflow raporlarını çalıştığınızda

features = tf.parse_single_example(
     serialized_example, 
     # Defaults are not specified since both keys are required. 
     features={ 
     ... # other stuff 
     'label': tf.FixedLenFeature(
      [4], dtype=tf.int64, default_value=-1)} 
    ) 
label = features['label'] 

: Açıkça

ValueError: Cannot reshape a tensor with 1 elements to shape [4] (4 elements) 

, ben

Answer 1

deneyin ayar varsayılan değer oldukça temel bir şey anlayamıyorum = [-1] * 4