[Generate the images pack tensor with the same random transform pattern] #python #pytorch

generate_images_pack_matrix.py

import numpy as np
import torch

def generate_images_pack_matrix(in_root, train=False):
    # make a seed with numpy generator 
    seed = np.random.randint(2147483647)
    
    in_root = Path(in_root)
    img_paths = [i for i in list(in_root.iterdir()) if i.is_file() and not i.name.startswith(".")]
    img_paths.sort(key=lambda x:int(x.stem))
    res=[]
    for img_path in img_paths:
        # apply this seed to img tranfsorms
        random.seed(seed)
        img = Image.open(str(img_path))
        if train:
            img = eye_train_transforms(img)
        else:
            img = eye_val_transforms(img)
        res.append(img)
    return torch.stack(res)