"""Implements a dataset class for handling image data"""
-from utils.image_utils import imread, imsave
DATA_PATH_BASE = '/home/VoxelFlow/dataset/ucf101_triplets/'
Args:
"""
self.data_list_file = data_list_file
- if process_func:
- self.process_func = process_func
- else:
- self.process_func = self.process_func
def read_data_list_file(self):
"""Reads the data list_file into python list
data_list = [DATA_PATH_BASE+line.rstrip() for line in f]
self.data_list = data_list
return data_list
-
- def process_func(self, example_line):
- """Process the single example line and return data
- Default behavior, assumes each line is the path to a single image.
- This is used to train a VAE.
- """
- return imread(example_line)