Error in training keypoints

DonsetPG / narya

The Narya API allows you track soccer player from camera inputs, and evaluate them with an Expected Discounted Goal (EDG) Agent. This repository contains the implementation of the flowing paper https://arxiv.org/abs/2101.05388. We also make available all of our pretrained agents, and the datasets we used as well.

MIT License

164 stars 48 forks source link

Error in training keypoints #36

Closed Rit-ctrl closed 3 years ago

Rit-ctrl commented 3 years ago

I faced tuple index out of range error while training the keypoints model line 111.

Here is the complete trace

https://gist.github.com/Rit-ctrl/13b448b1ac115297d1abf43db9f18eed

I get this error irrespective of the dataset I use for training

DonsetPG commented 3 years ago

Hey, can it be linked with issues#35 ?

Could you try the solutions given by @alen1010 and see if this solve the issue? Otherwise I'll try to investigate.

Rit-ctrl commented 3 years ago

Hey, can it be linked with issues#35 ?

Yeah sure.

Could you try the solutions given by @alen1010 and see if this solve the issue? Otherwise I'll try to investigate.

Yeah I tried them. Still the same error arises

DonsetPG commented 3 years ago

I can't reproduce your error on my side. However, using solutions proposed in issues#35 and modifying class Dataloder(keras.utils.Sequence): to tf.keras.utils.Sequence let's me train the model.

Rit-ctrl commented 3 years ago

Would it be possible for you to share the Colab file which you used for testing now?