Helpers Module

ARGS

class helpers.ARGS.ARGS(self, model, dataset_path, num_classes, labels_type, batch_size=8, epochs=80)

[source]

Parameters

• model (str) - Model name in available models list. [list]
• dataset_path (str) - Path to the dataset root directory.
• num_classes (int) - Number of classes in the dataset.
• labels_type (str) - Label format in the dataset (csv, json or label (images)).
• batch_size (int) - Batch size for dataloaders (batch_size != 1).
• epochs (int) - Number of epochs for training.

Returns

• args

Return type

• helpers.ARGS.ARGS