| from fast_pointnet_v2 import train_pointnet | |
| if __name__ == "__main__": | |
| # Load the dataset | |
| dataset_path = "path/to/your/dataset" | |
| model_save_path = "path/to/your/model.pth" | |
| # Train the model | |
| train_pointnet(dataset_path, model_save_path, epochs=100, batch_size=512, lr=0.001, score_weight=0.25, class_weight=1.0) |