git.sesse.net Git - voxel-flow/blob - dataset.py

   1 """Implements a dataset class for handling image data"""
   2
   3 DATA_PATH_BASE = '/home/VoxelFlow/dataset/ucf101_triplets/'
   4
   5 class Dataset(object):
   6   def __init__(self, data_list_file=None, process_func=None):
   7     """
   8       Args:
   9     """
  10     self.data_list_file = data_list_file
  11
  12   def read_data_list_file(self):
  13     """Reads the data list_file into python list
  14     """
  15     f = open(self.data_list_file)
  16     data_list =  [DATA_PATH_BASE+line.rstrip() for line in f]
  17     self.data_list = data_list
  18     return data_list