Spaces:

DocForg
/

Document_Forgery_Detection

Sleeping

Document_Forgery_Detection / src /data /__init__.py

Upload 54 files

ff0e79e verified 10 months ago

522 Bytes

"""Data module"""

from .preprocessing import DocumentPreprocessor, preprocess_image
from .augmentation import DatasetAwareAugmentation, get_augmentation
from .datasets import (
    DocTamperDataset,
    RTMDataset,
    CASIADataset,
    ReceiptsDataset,
    get_dataset
)

__all__ = [
    'DocumentPreprocessor',
    'preprocess_image',
    'DatasetAwareAugmentation',
    'get_augmentation',
    'DocTamperDataset',
    'RTMDataset',
    'CASIADataset',
    'ReceiptsDataset',
    'get_dataset'
]

	"""Data module"""

	from .preprocessing import DocumentPreprocessor, preprocess_image
	from .augmentation import DatasetAwareAugmentation, get_augmentation
	from .datasets import (
	DocTamperDataset,
	RTMDataset,
	CASIADataset,
	ReceiptsDataset,
	get_dataset
	)

	__all__ = [
	'DocumentPreprocessor',
	'preprocess_image',
	'DatasetAwareAugmentation',
	'get_augmentation',
	'DocTamperDataset',
	'RTMDataset',
	'CASIADataset',
	'ReceiptsDataset',
	'get_dataset'
	]