lack of func "make_one_shot_iterator()""get_next()",so the train_gt/test_gt can't be got.
in train.py ,there is
"with tf.name_scope('dataset'):
train_loader = DataLoader(train_folder, resize_height=height, resize_width=width)
train_dataset = train_loader(batch_size=batch_size, time_steps=num_his, num_pred=1)
train_it = train_dataset.make_one_shot_iterator()
train_videos_clips_tensor = train_it.get_next()
train_videos_clips_tensor**.set_shape**([batch_size, height, width, 3*(num_his + 1)])
train_inputs = train_videos_clips_tensor[..., 0:num_his*3]
train_gt = train_videos_clips_tensor[..., -3:]
print('train inputs = {}'.format(train_inputs))
print('train prediction gt = {}'.format(train_gt))
test_loader = DataLoader(test_folder, resize_height=height, resize_width=width)
test_dataset = test_loader(batch_size=batch_size, time_steps=num_his, num_pred=1)
test_it = test_dataset.make_one_shot_iterator()
test_videos_clips_tensor = test_it.get_next()
test_videos_clips_tensor.set_shape([batch_size, height, width, 3*(num_his + 1)])
test_inputs = test_videos_clips_tensor[..., 0:num_his*3]
test_gt = test_videos_clips_tensor[..., -3:]"
however,there is no "make_one_shot_iterator()""get_next()""set_shape" in whole project. Could you please add the above code?
lack of func "make_one_shot_iterator()""get_next()",so the train_gt/test_gt can't be got.
in train.py ,there is
"with tf.name_scope('dataset'):
train_loader = DataLoader(train_folder, resize_height=height, resize_width=width)
train_dataset = train_loader(batch_size=batch_size, time_steps=num_his, num_pred=1)
train_it = train_dataset.make_one_shot_iterator()
train_videos_clips_tensor = train_it.get_next()
train_videos_clips_tensor**.set_shape**([batch_size, height, width, 3*(num_his + 1)])
train_inputs = train_videos_clips_tensor[..., 0:num_his*3]
train_gt = train_videos_clips_tensor[..., -3:]
print('train inputs = {}'.format(train_inputs))
print('train prediction gt = {}'.format(train_gt))
test_loader = DataLoader(test_folder, resize_height=height, resize_width=width)
test_dataset = test_loader(batch_size=batch_size, time_steps=num_his, num_pred=1)
test_it = test_dataset.make_one_shot_iterator()
test_videos_clips_tensor = test_it.get_next()
test_videos_clips_tensor.set_shape([batch_size, height, width, 3*(num_his + 1)])
test_inputs = test_videos_clips_tensor[..., 0:num_his*3]
test_gt = test_videos_clips_tensor[..., -3:]"
however,there is no "make_one_shot_iterator()""get_next()""set_shape" in whole project. Could you please add the above code?