pointnet2-tensorflow2 icon indicating copy to clipboard operation
pointnet2-tensorflow2 copied to clipboard

Training my own dataset (radar point dataset) failed

Open ghost opened this issue 5 years ago • 0 comments

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!

ghost avatar Sep 27 '20 18:09 ghost