Skip to content

Instantly share code, notes, and snippets.

View qiaopengju's full-sized avatar

PengJu Qiao qiaopengju

View GitHub Profile
import h5py
import helpers
import numpy as np
from pathlib import Path
import torch
from torch.utils import data
class HDF5Dataset(data.Dataset):
"""Represents an abstract HDF5 dataset.