Spaces:
Running on Zero
Running on Zero
| #!/usr/bin/env python3 | |
| """ | |
| Utilities for Training Sample Generation | |
| This package contains utility modules for generating training samples and doing preprocessing for RAP. | |
| """ | |
| # Import commonly used functions for easier access | |
| from .submap_utils import ( | |
| get_default_num_samples, | |
| create_submap_from_frames, | |
| generate_submap_boundaries_for_sample, | |
| select_spatially_close_submaps, | |
| validate_no_overlap | |
| ) | |
| from .io_utils import ( | |
| create_metadata_json, | |
| convert_to_hdf5, | |
| get_dataset_name, | |
| load_sample_from_folder, | |
| load_sample_from_hdf5, | |
| save_processed_sample, | |
| save_training_sample | |
| ) | |
| from .processing_utils import ( | |
| process_nss_dataset, | |
| process_nss_multi_dataset, | |
| process_sequence_with_loader, | |
| set_random_seeds, | |
| process_tls_dataset | |
| ) | |
| from .validation_utils import ( | |
| perform_dry_run, | |
| process_sequences, | |
| _validate_and_setup_args | |
| ) | |
| from .split_utils import ( | |
| create_data_splits, | |
| create_nss_data_splits, | |
| create_random_data_splits_only, | |
| create_sequence_data_splits_only, | |
| copy_and_update_data_split | |
| ) | |
| from .preview_utils import ( | |
| preview_data_splits | |
| ) | |
| from .feature_extraction_metadata_utils import ( | |
| save_processing_metadata, | |
| print_detailed_statistics | |
| ) | |
| from .point_sampling_utils import ( | |
| calculate_voxel_coverage, | |
| calculate_adaptive_sample_count_per_part, | |
| allocate_fps_points, | |
| apply_batched_fps | |
| ) | |
| from .voxel_utils import ( | |
| compute_adaptive_voxel_size, | |
| ) | |
| # Import dataset utilities for easier access | |
| from . import dataset_utils | |
| __all__ = [ | |
| # Submap utilities | |
| 'get_default_num_samples', | |
| 'create_submap_from_frames', | |
| 'generate_submap_boundaries_for_sample', | |
| 'select_spatially_close_submaps', | |
| 'validate_no_overlap', | |
| # I/O utilities | |
| 'create_metadata_json', | |
| 'convert_to_hdf5', | |
| 'get_dataset_name', | |
| 'load_sample_from_folder', | |
| 'load_sample_from_hdf5', | |
| 'save_processed_sample', | |
| 'save_training_sample', | |
| # Processing utilities | |
| 'process_nss_dataset', | |
| 'process_nss_multi_dataset', | |
| 'process_sequence_with_loader', | |
| 'set_random_seeds', | |
| 'process_tls_dataset', | |
| # Validation utilities | |
| 'perform_dry_run', | |
| 'process_sequences', | |
| '_validate_and_setup_args', | |
| # Split utilities | |
| 'create_data_splits', | |
| 'create_nss_data_splits', | |
| 'create_random_data_splits_only', | |
| 'create_sequence_data_splits_only', | |
| 'copy_and_update_data_split', | |
| # Preview utilities | |
| 'preview_data_splits', | |
| # Feature Extraction Metadata Utilities | |
| 'save_processing_metadata', | |
| 'print_detailed_statistics', | |
| # Point Sampling Utilities | |
| 'calculate_voxel_coverage', | |
| 'calculate_adaptive_sample_count_per_part', | |
| 'allocate_fps_points', | |
| 'apply_batched_fps', | |
| # Voxel utilities | |
| 'compute_adaptive_voxel_size', | |
| # Dataset utilities | |
| 'dataset_utils', | |
| 'save_num_points_to_folder', | |
| 'save_points_to_ply' # Explicitly expose save_points_to_ply | |
| ] |