提交 b0a84e61 编写于 作者: S SunGaofeng

remove redundant codes and comments

上级 b0888e63
......@@ -89,7 +89,7 @@ class NonlocalReader(DataReader):
**dataset_args)
else:
logger.info('Not implemented')
raise
raise NotImplementedError
def video_fast_get_frame(video_path,
......@@ -104,7 +104,6 @@ def video_fast_get_frame(video_path,
sampledFrames = []
# n_frame < sample area
video_output = np.ndarray(shape=[length, height, width, 3], dtype=np.uint8)
use_start_frm = start_frm
......@@ -220,7 +219,6 @@ def make_reader(filelist, batch_size, sample_times, is_training, shuffle,
start_frm = -1
spatial_pos = -1
in_sample_times = 1
# print('label = ', label)
label = np.array([label]).astype(np.int64)
# 1, get rgb data for fixed length of frames
try:
......@@ -246,8 +244,6 @@ def make_reader(filelist, batch_size, sample_times, is_training, shuffle,
spatial_pos = spatial_pos)
batch_out.append((rgbdata, label))
#elapsed_time = time.time() - start_time
#print('read item time ', elapsed_time)
if len(batch_out) == batch_size:
yield batch_out
batch_out = []
......
......@@ -34,7 +34,7 @@ class AttentionCluster(ModelBase):
self.feature_dims = self.cfg.MODEL.feature_dims
self.cluster_nums = self.cfg.MODEL.cluster_nums
self.seg_num = self.cfg.MODEL.seg_num
self.class_num = self.cfg.MODEL.num_classes #self.cfg.MODEL.class_num
self.class_num = self.cfg.MODEL.num_classes
self.drop_rate = self.cfg.MODEL.drop_rate
if self.mode == 'train':
......
......@@ -46,11 +46,7 @@ f.close()
def generate_pkl(entry):
mode = entry[4]
category = entry[0].strip('"')
#if ' ' in category:
# category_dir = category.replace(' ', '_')
#else:
category_dir = category
#video_path = os.path.join(entry[0], entry[1] + "_%06d"%int(entry[2]) + "_%06d"%int(entry[3]) + ".mp4")
video_path = os.path.join(
'./',
entry[1] + "_%06d" % int(entry[2]) + "_%06d" % int(entry[3]) + ".mp4")
......
......@@ -142,11 +142,6 @@ class YT8MFrameFeatureReader(BaseReader):
num_frames = tf.minimum(tf.shape(decoded_features)[0], max_frames)
#feature_matrix = Dequantize(decoded_features,
# max_quantized_value,
# min_quantized_value)
#feature_matrix = resize_axis(feature_matrix, 0, max_frames)
feature_matrix = decoded_features
return feature_matrix, num_frames
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册