Skip to content

Conversation

lgleznah
Copy link

@lgleznah lgleznah commented Apr 2, 2025

This PR addresses multiple issues I have found during my own experiments with this repo. Hopefully, these will be helpful for those who are working with this technology, which so far, looks promising for doing point cloud stuff.

This is a quick summary of all the stuff I have added:

  • Validation max epochs is now interpreted as stated.
  • Checkpoints are now saved of the current best model in terms of mean IoU.
  • Option for specifying class weights by setting cfg.data.compute_class_weights = True. Addresses Imbalance datasets tackling #9
  • Training and validation losses visualization for monitoring potential overfitting during training.
  • Validation step now shows confusion matrix.
  • Testing script now generates a PLY file with the predictions, for visualization purposes.
  • Other minor bugfixes.

lgleznah added 8 commits April 2, 2025 09:52
Testing script now works when there is no subsampling. Prediction results are now returned, for visuaization purposes
Trainval script now keeps track of the best model in terms of mean IoU. Furthermore, training and validation losses are now returned from their respective scripts, in order to visualize their evolution during training. Finally, checkpoints now keep track of last epoch, in order to properly resume training.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant