| from fast_pointnet import train_pointnet | |
| import os | |
| if __name__ == "__main__": | |
| # Load the dataset | |
| dataset_path = "/home/skvrnjan/personal/hohocustom/" | |
| model_save_path = "/home/skvrnjan/personal/hoho_pnet/" | |
| os.makedirs(model_save_path, exist_ok=True) | |
| # Train the model | |
| train_pointnet(dataset_path, model_save_path) |