Skip to content

Commit ec681bb

Browse files
committed
Fix:FILE IO TD/T1
1 parent 5e3d27e commit ec681bb

1 file changed

Lines changed: 18 additions & 2 deletions

File tree

src/io_modules/file_reader.py

Lines changed: 18 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -261,6 +261,21 @@ def read_file_info(self):
261261
def skipHeader(self):
262262
self.filehandle.seek(self.header_size,0)
263263

264+
def initialize_read(self,makeShared=True):
265+
''' Initialize the file_reader options'''
266+
logger.debug("File reader is initialzed again for file reading");
267+
self.finished = False
268+
if self.partition_num == 1:
269+
logger.debug("File reader:Only one partition keep same.");
270+
else:
271+
self.num_pad_frames = 0
272+
#self.frames_per_partition = 0
273+
self.partition_num = 0
274+
self.label = None
275+
self.feat = None
276+
self.skipHeader()
277+
#self.read_next_partition_data(makeShared=makeShared)
278+
264279
def read_next_partition_data(self,already_read=0,pad_zeros=False,makeShared=True):
265280

266281
if self.finished:
@@ -324,6 +339,7 @@ def __init__(self,path,options):
324339
self.options=options;
325340
self.batch_size = options['batch_size']
326341
self.partition = options['partition']
342+
self.frames_per_class=0;
327343
self.read_file_info();
328344

329345
def read_file_info(self):
@@ -422,10 +438,10 @@ def read_next_partition_data(self,already_read=0,pad_zeros=False,makeShared=True
422438
logger.debug('TD Filereader : NO more frames to read from %s',self.filepath)
423439

424440
def initialize_read(self,makeShared=True):
425-
logger.debug("File reader is initialzed again for file reading");
441+
logger.debug("File reader [T1] is initialzed again for file reading");
426442
self.finished = False
427443
if self.partition_num == 1:
428-
logger.debug("File reader:Only one partition keep same.");
444+
logger.debug("File reader [T1]:Only one partition. keep same.");
429445
else:
430446
self.num_pad_frames = 0
431447
#self.frames_per_partition = 0

0 commit comments

Comments
 (0)