search

Sense-X / UniFormer

[ICLR2022] official implementation of UniFormer
Apache License 2.0
828 stars 111 forks source link

How to custom train of unifomer for video classification on my own custom dataset #136

Open AbrarKhan009 opened 4 months ago

AbrarKhan009 commented 4 months ago

Hello everyone,

I would like some guidance on how to train UniFormer using my custom video dataset. I have structured my dataset similarly to the Kinetics-400 dataset, with separate train and validation folders, as well as corresponding CSV files for both.

My goal is to fine-tune a UniFormer model that has been pre-trained on the Kinetics-400 dataset using my own data. I would greatly appreciate any suggestions or detailed steps on how to achieve this.

Thank you in advance for your assistance!

Andy1621 commented 2 months ago

Sorry for missed response. You can follow the scripts used for UCF101 to train your own data.