CCDeep.tools package

Submodules

CCDeep.tools.augmentation module

class CCDeep.tools.augmentation.Augmentor(image: ndarray)

基类：object

adjustBright(gamma): gamma > 1, bright; gamma < 1. dark

flipHorizontal()

flipVertical()

movePosition()

raw()

rotate(angle)

CCDeep.tools.augmentation.augment(dic, mcy)

CCDeep.tools.augmentation.augment_before_train(raw_folder, saved_folder): 如果拥有足够的存储空间，或者内存有限，请在开始训练模型之前，使用此函数进行数据增强，并且将配置文件中的AUGMENTATION_IN_TRAINING 设置为False。否则，请不要调用次函数，并且将AUGMENTATION_IN_TRAINING设置为True。 :param raw_folder: 原始数据所在目录 :param saved_folder: 通过数据增强生成的文件保存目录

CCDeep.tools.augmentation.augment_in_train(img, label)

CCDeep.tools.generate_dataset module

CCDeep.tools.split_dataset module

class CCDeep.tools.split_dataset.SplitDataset(dataset_dir, saved_dataset_dir, train_ratio=0.7, test_ratio=0.15, show_progress=False)

基类：object

start_splitting()

CCDeep.tools.split_dataset.split(dataset_mcy_path, save_mcy_path)

Module contents