Function get_data_iterator

src/utils/triplane_util.py:64–69

(featmaps_data, sizes, batch_size=1)

def get_data_iterator(featmaps_data, sizes, batch_size=1):
    featmaps_data = featmaps_data.unsqueeze(0).expand(batch_size, -1, -1, -1)
    H, W, D = sizes

    while True:
        yield featmaps_data, {'H': H, 'W': W, 'D': D}