feiyuhuahuo / Anomaly_Prediction

Pytorch implementation of anomaly prediction.
78 stars 22 forks source link

怎么去改变提取的预测帧数呢? #30

Open TPF0628 opened 3 months ago

TPF0628 commented 3 months ago

对于我自己的数据集我想要修改为输入10帧,通过前9帧来预测下一帧图片。但是在Dataset中修改了 self.clip_length = 10 random_seq = list(range(len(all_imgs) - 9))之后训练运行到640轮都会报一个错误 IndexError: list index out of range 具体发生在 video_clip.append(np_load_frame(one_folder[i], self.img_h, self.img_w)) 通过打印信息应该是取到的start超过了范围。 请问我该怎么解决呢?