提交 6da62e65 编写于 作者: H huangjun12 提交者: SunGaofeng

add log out for tall reader (#3825)

* add log out for tall reader

* fix log info of tall reader
上级 4bb3163d
......@@ -144,7 +144,7 @@ class TALLReader(DataReader):
for sent_vec in sent_vecs:
self.clip_sentence_pairs.append((clip_name, sent_vec)) #10146
logger.info(self.mode.upper() + ':' + str(
len(self.clip_sentence_pairs)) + " clip-sentence pairs are readed")
len(self.clip_sentence_pairs)) + " clip-sentence pairs are read")
movie_names_set = set()
movie_clip_names = {}
......@@ -167,7 +167,12 @@ class TALLReader(DataReader):
# TALL model doesn't take validation during training, it will test after all the training epochs finish.
return
if self.mode == 'train':
num_sliding_clips = len(sliding_clips_tmp)
count = 0
for clip_name in sliding_clips_tmp:
count += 1
logger.info('processing data ............' + str(count) + '/' +
str(num_sliding_clips))
if clip_name.split(".")[2] == "npy":
movie_name = clip_name.split("_")[0]
for clip_sentence in self.clip_sentence_pairs:
......@@ -193,7 +198,7 @@ class TALLReader(DataReader):
(clip_sentence[0], clip_sentence[1],
clip_name, start_offset, end_offset))
logger.info('TRAIN:' + str(len(self.clip_sentence_pairs_iou)) +
" iou clip-sentence pairs are readed")
" iou clip-sentence pairs are read")
elif (self.mode == 'test') or (self.mode == 'infer'):
for clip_name in sliding_clips_tmp:
......@@ -205,7 +210,7 @@ class TALLReader(DataReader):
[1])
logger.info('TEST:' + str(len(self.clip_sentence_pairs_iou)) +
" iou clip-sentence pairs are readed")
" iou clip-sentence pairs are read")
def load_movie_slidingclip(self, clip_sentence_pairs,
clip_sentence_pairs_iou, movie_name):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册