Skip to content

Training my own dataset (radar point dataset) failed #8

@ghost

Description

Hi,

I'd like first of all to thank you for publishing your code, it's very helpful. I am currently trying to train my own dataset using PointNet++ and semantic segmentation model. I tried to convert my data(data that I split myself into training and validation )to tf record.
I have labels for training and validation and also points for training and validation, both into arrays.
Labels is an np array containing the indexes of labels of type int64 and Points is an np.array containing points corresponding to each label of type float32

This way:

def tables_to_TF( Label,Points,tf_filename):
    
    filepath = os.path.join(tf_filename)
    print('Writing', filepath)
    writer = tf.io.TFRecordWriter(tf_filename)
    example = tf.train.Example(
            features=tf.train.Features(feature={
            'Points': _float_feature(Points),
            'Labels': _int64_feature(Label),
            
                }) 
                    )
    
    writer.write(example.SerializeToString())

But loading the data does not work, I get can't sparse serialized Example.
Now when trying to train the dataset using your code I get this:

image

I don't understand why it does not work and why the conversion doesn't work. I'm stuck and I have tried everything but could not resolve the problem.

It will be very kind of you if you could help me. I'm counting on your help.

Thank you in advance.

Kind regards!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions