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
166 stars 48 forks source link

Colab crashes when running import TrackingDatasetBuilder #41

Open Rit-ctrl opened 3 years ago

Rit-ctrl commented 3 years ago

I tried to run tracker_train in colab with GPU enabled. While running this line

from narya.datasets.tracking_dataset import TrackingDatasetBuilder my colab crashes.