ZhengChang467 / STRPM

STRPM: A Spatiotemporal Residual Predictive Model for High-Resolution Video Prediction, CVPR2022
MIT License
17 stars 3 forks source link

How to construct the train data? #2

Open chufall opened 2 years ago

chufall commented 2 years ago

Dear authors of STRPM,

I'm trying to train my own dataset with your code.

From you code: print('Loading train dataset') self.path = data_train_path with codecs.open(self.path) as f: self.file_list = f.readlines() print('Loading train dataset finished, with size:', len(self.file_list)) I think the content of train file should like: data/ucfsport/Kicking-Side/,6 data/ucfsport/Kicking-Side/,7 data/ucfsport/Kicking-Side/,8 data/ucfsport/Kicking-Side/,9

the first line means using 6-9th to predict 10th the second line means using 7-10th to predict 11th and so on.

Am I right? I appreciate your reply!

Thanks a lot! Chu

ZhengChang467 commented 2 years ago

Right!