jfzhang95 / pytorch-video-recognition

PyTorch implemented C3D, R3D, R2Plus1D models for video activity recognition.
MIT License
1.16k stars 250 forks source link

how can I do transfer learning with my dataset ? #6

Closed bowin closed 5 years ago

bowin commented 5 years ago

Please help

jfzhang95 commented 5 years ago
  1. You can write your own dataloader to load your dataset.
  2. Change num_classes to fit your dataset's labels in here.