Προσπαθώ να δημιουργήσει tfrecords με τον ίδιο τρόπο FSNS δημιουργήθηκε;

ψήφοι
0

όταν τρέχει αυτό το σενάριο:

def _int64_feature(value):
    return tf.train.Feature(int64_list=tf.train.Int64List(value=value))
for j in range(0,int(len(addrs_image))):
    print('Train data: {}/{}'.format(j,int(len(addrs_image))))
    sys.stdout.flush()

    img = Image.open(addrs_image[j])

    img = img.resize((600, 150), Image.ANTIALIAS)
    np_data = np.array(img)
    image_data = img.tobytes()
    for text in open(addrs_label[j], encoding=utf-8):
        char_ids_padded, char_ids_unpadded = encode_utf8_string(
                            text=text,
                            dic=dict,
                            length=7,
                            null_char_id=26)
    example = tf.train.Example(features=tf.train.Features(
                        feature={
                            'image/encoded': _bytes_feature(image_data),
                            'image/format': _bytes_feature(braw),
                            'image/width': 
                             _int64_feature([np_data.shape[1]]),
                             'image/orig_width': 
                              _int64_feature([np_data.shape[1]]),
                            'image/class': _int64_feature(char_ids_padded),
                            'image/unpadded_class': 
                              _int64_feature(char_ids_unpadded),
                            'image/text': _bytes_feature(bytes(text, 'utf-8')),
                            # 'height': _int64_feature([crop_data.shape[0]]),
                        }
                    ))

Πήρα ένα σφάλμα που λέει: Τύπος σφάλματος: 'image/class': _int64_feature(char_ids_padded), 'image/unpadded_class': _int64_feature(char_ids_unpadded), r: κανένας δεν έχει πληκτρολογήστε nonetype και την έξοδο αναμένεται να είναι int

και το αρχείο dic.txt περιέχει εισάγετε περιγραφή της εικόνας εδώ

Έτσι, κάθε ένα έχει ιδέα για το πώς να λύσει αυτό το πρόβλημα; οποιαδήποτε πρόταση είναι πολύ εκτίμησα και σας ευχαριστώ όλους εκ των προτέρων

Δημοσιεύθηκε 27/11/2018 στις 17:36
πηγή χρήστη
Σε άλλες γλώσσες...                            

Cookies help us deliver our services. By using our services, you agree to our use of cookies. Learn more