Documentation for `balance_dataset`

Path: image_classification/balance_dataset.py

Module Description

Script to rebalance the site classification dataset by redistributing images across train, validation, and test sets to ensure all classes are represented.

Functions

`parse_arguments`

No description provided.

`collect_all_images`

Collect all images from the input directory, organized by class.

`split_images_by_class`

Split images for each class according to the specified ratios.

`create_output_structure`

Create the output directory structure and copy files.

`print_dataset_summary`

Print a summary of the rebalanced dataset.

`main`

No description provided.

Documentation for balance_dataset ​

Module Description ​

Functions ​

parse_arguments ​

collect_all_images ​

split_images_by_class ​

create_output_structure ​

print_dataset_summary ​

main ​