You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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?
The text was updated successfully, but these errors were encountered:
sjzhang-Gt
changed the title
Cannot train because of the lack of func "make_one_shot_iterator()",so the train_gt/test_gt can't be got
Cannot train because of the lack of func "make_one_shot_iterator()"
May 10, 2024
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?
The text was updated successfully, but these errors were encountered: