yanx27 / Pointnet_Pointnet2_pytorch

PointNet and PointNet++ implemented by pytorch (pure python) and on ModelNet, ShapeNet and S3DIS.
MIT License
3.64k stars 889 forks source link

Don't work for different dataset (custom data)? #154

Open AI-Hunter opened 2 years ago

AI-Hunter commented 2 years ago

When I use S3DIS dataset, it is working, but when I am using my own dataset, it is stuck at classifier.train()! What is the reason? is it normalization issue?

I converted into .npy from my las file using the collect_indoor3d_data.py ( a little modified base on my class number), and I don't have rgb values, so I have set the rgb values with 255.

Is there any suggestion please?

gang3217 commented 2 years ago

Have you solved this problem yet

wyz5008 commented 2 years ago

你好,我也在尝试做自己的语义分割数据集,(已经采集做好了PCD格式的点云文件),请问你是如何打语义分割的标签,并且调整格式以及修改代码使其可以训练呢?

bhagatdas commented 2 years ago

Hi... @AI-Hunter @gang3217 @wyz5008 @lly007 @yushiangw anyone solved the problem for custom dataset.. I am also stuct at BN momentum updated to: 0.100000 0% 0/6 [00:00<?, ?it/s] or classifier = classifier.train() please help

sean880304 commented 1 year ago

@bhagatdas https://github.com/yanx27/Pointnet_Pointnet2_pytorch/blob/eb64fe0b4c24055559cea26299cb485dcb43d8dd/data_utils/S3DISDataLoader.py#L59 change this number to a smaller one (say 120), this may resolve your problem.

bhagatdas commented 1 year ago

@sean880304 Yeah.. Did the same while ago... now it is working well with the custom dataset.

foysalahmed10 commented 1 year ago

@sean880304 @bhagatdas First, I would like to thank you for your time and support. I'm trying to train my own dataset. Can you tell me how can I prepare my dataset for training, step by step. I already completed labeling, now the dataset in .PCD and .txt format.

bhagatdas commented 1 year ago

I used the S3DIS format to train my own data. So, first go through the S3DIS data format, run the mode using it. after that you ll know how to convert your own dataset to use the model.

foysalahmed10 commented 1 year ago

@bhagatdas Thank you so much for your reply. I will follow your advice.

Akshatha-Jagadish commented 1 year ago

@bhagatdas could you please tell why the point cloud data is not normalized in s3dis data loader? Thank you

bhagatdas commented 1 year ago

@Akshatha-Jagadish I don't have any particular answer, but if you are doing it, let me know the result. I was getting around 88% accuracy with Toronto-3D dataset.