PET_img_loader.py

import h5py

##############################################################################
# Script for importing images created from PET-scanning of mice.
# Data is returend in a shape suitable for CNNs in Pytorch.
##############################################################################


def data_loader(path):
    # Function for importing data in raw form.

    # path: Path to stored data.
    
    with h5py.File(path, 'r') as file:
        a = list(file['I'])
    return hdf5storage.loadmat(path)['I']