search

A4Bio / SimVP

The official implementation of the CVPR'22 paper SimVP: Simpler Yet Better Video Prediction.
205 stars 40 forks source link

Custom dataset generation #13

Open stardustLi opened 9 months ago

stardustLi commented 9 months ago

Can you give me instructions on how to train the network on a custom dataset? It contains 512x512 RGB images with labels in the form of timestamps of when they were taken, in differences of minutes. I'm trying to predict what the next images in the sequence will look like in the next timestamps.

chengtan9907 commented 9 months ago

Thanks for your interest! Maybe you can refer to the tutorial of OpenSTL: OpenSTL-Tutorial.

SamuelSchmidgall commented 8 months ago

@stardustLi I wrote code that did this for my own dataset if you're wanting to know more just msg me. Specifically, I have a very large dataset so I wrote a lazy mp4 loader that reads and loads the files on demand which might be slower than what you need..