How to train it on our own custom data?

wmcnally / golfdb

GolfDB is a video database for Golf Swing Sequencing, which involves detecting 8 golf swing events in trimmed golf swing videos. This repo demos the baseline model, SwingNet.

198 stars 60 forks source link

How to train it on our own custom data? #17

Open st186 opened 1 year ago

st186 commented 1 year ago

How can I train my model on custom videos that I have with me? Can you help me out?

AlecSkril commented 12 months ago

I also have this question. I am trying to train on customer data that is a different size than the 160x160. I am getting an error with transforms.Compose(...) (line 34 in train.py). I will play around with the tensor size but if anyone has any suggestions, I would appreciate it. Thanks.

AlecSkril commented 12 months ago

@wmcnally