search

IsaacGuan / PointNet-Plane-Detection

A 3D plane detection approach using PointNet
GNU General Public License v3.0
99 stars 25 forks source link

My model segments all points in one class #11

Open Nhunguts opened 3 years ago

Nhunguts commented 3 years ago

Hi IsaacGuan @IsaacGuan Thank you for publishing your code. I have used your code to train my own data. My data also has two classes like yours. However, my model can't segment any positive point. It segmented all points in one class.
What should I check now? Thank you.

BXESYR commented 3 years ago

You can check the parameter Is_trained is True or False.

Nhunguts commented 3 years ago

You can check the parameter Is_trained is True or False.

Hi BXESYR, in the train.py file, "is_training=True" in "def train_one_epoch(epoch_number)" and "is_training=True" in "def eval_one_epoch(epoch_number). is this right? Or do you mean something else need to edit? Please help, thank you.