Spaces:

reysarms
/

ccs_229_final_project

Sleeping

ccs_229_final_project / utils /preprocessing.py

updated

c08ab4e over 1 year ago

348 Bytes

from torchvision import transforms

def get_transforms():
    transform = transforms.Compose([
        transforms.Resize((224, 224)),
        transforms.ToTensor(),
        transforms.Normalize(
            mean=[0.485, 0.456, 0.406],  # ImageNet means
            std=[0.229, 0.224, 0.225]    # ImageNet stds
        )
    ])
    return transform

	from torchvision import transforms

	def get_transforms():
	transform = transforms.Compose([
	transforms.Resize((224, 224)),
	transforms.ToTensor(),
	transforms.Normalize(
	mean=[0.485, 0.456, 0.406], # ImageNet means
	std=[0.229, 0.224, 0.225] # ImageNet stds
	)
	])
	return transform