My part segmentation model segments all points in one class

charlesq34 / pointnet

PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation

Other

4.64k stars 1.44k forks source link

My part segmentation model segments all points in one class #267

Open Nhunguts opened 3 years ago

Nhunguts commented 3 years ago

Hi @charlesq34 and users, Thank you, Charlesq, for sharing your code. I am trying to use/modify your code for binary segmentation.
However, in the training phase, the accuracy didn't increase. In the testing phase, the model segments all points in one class. Could you please help me what should I change in the code? Thank you. total_mean_loss

chongma commented 2 years ago

I am getting same result. Did you find a solution?

Nhunguts commented 2 years ago

Hi @chongma , I am sorry that I haven't found a solution for this. Charles said to me that this is a low-level error, but I still don't know how to fix it.