search

mahmoodlab / UNI

Towards a general-purpose foundation model for computational pathology - Nature Medicine
Other
348 stars 48 forks source link

Transform #33

Closed JSluo888 closed 5 months ago

JSluo888 commented 5 months ago

I am wondering if we have to resize to 224, and normalize using the parameters in the readme? If not, how should we determine the values of these parameters?

transform = transforms.Compose(
    [
        transforms.Resize(224),
        transforms.ToTensor(),
        transforms.Normalize(mean=(0.485, 0.456, 0.406), std=(0.229, 0.224, 0.225)),
    ]
)
lsnls commented 1 month ago

Hello, I have the same doubt. Please have you solved it？