ecopulse / src /transforms.py
acibZ's picture
Deploy EcoPulse
43abac3
Raw
History Blame Contribute Delete
404 Bytes
"""
Shared image transformations for EcoPulse.
Consolidates the identical transforms.Compose block that was copy-pasted
across multiple modules into a single importable constant.
"""
from torchvision import transforms
EUROSAT_TRANSFORM = transforms.Compose([
transforms.Resize((64, 64)),
transforms.ToTensor(),
transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
])