Как я могу дать изображения и метки для DataLoader в Pytorch? - PullRequest
1 голос
/ 01 апреля 2019

У меня есть 2D-тензор с информацией о моих кадрах и 1D-тензор с информацией о выходе или метке, связанной с каждым кадром. Я делаю функцию и возвращаю (кадры, метки). Затем я передаю эти два параметра "torch.utils.data.DataLoader ()". Но DataLoader не работает для обоих (фреймы, метки) и просто может выполнить итерацию для одного из них.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...